Coding-python-标签-labelme_json2txt

代码

import os
import json

def convert_labelme_to_yolo(json_file, output_folder, class_dict):
    # Read the JSON file
    with open(json_file, 'r') as f:
        data = json.load(f)

    # Get image dimensions
    img_width = data['imageWidth']
    img_height = data['imageHeight']

    # Get the image name without extension
    img_name = os.path.splitext(os.path.basename(json_file))[0]

    # Create YOLO format label file
    txt_filename = os.path.join(output_folder, img_name + '.txt')
    with open(txt_filename, 'w') as f:
        for shape in data['shapes']:
            if shape['shape_type'] != 'rectangle':
                continue

            # Get category
            label = shape['label']
            class_id = class_dict[label]

            # Get bounding box coordinates
            points = shape['points']
            xmin, ymin = points[0]
            xmax, ymax = points[1]

            # Convert to YOLO format required coordinates
            x_center = (xmin + xmax) / 2.0 / img_width
            y_center = (ymin + ymax) / 2.0 / img_height
            width = (xmax - xmin) / img_width
            height = (ymax - ymin) / img_height

            # Write to label file
            line = f"{class_id} {x_center:.6f} {y_center:.6f} {width:.6f} {height:.6f}\n"
            f.write(line)


def main():
    # Define your class dictionary, where keys are class names and values are class IDs
    # class_dict = {'Brassiere': 14, 'panties': 15}

    class_dict = {
    'Short sleeve top': 1,
    'Long sleeve top': 2,
    'Short sleeve outwear': 3,
    'Long sleeve outwear': 4,
    'Vest': 5,
    'Sling': 6,
    'Shorts': 7,
    'Trousers': 8,
    'Skirt': 9,
    'Short sleeve dress': 10,
    'Long sleeve dress': 11,
    'Vest dress': 12,
    'Sling dress': 13,
}


    # Specify the folder containing JSON files and the output folder for YOLO labels
    input_folder = 'train\\annos'
    output_folder = 'train\\lables'

    # Iterate over all JSON files in the input folder
    for filename in os.listdir(input_folder):
        if filename.endswith('.json'):
            json_file = os.path.join(input_folder, filename)
            convert_labelme_to_yolo(json_file, output_folder, class_dict)

if __name__ == "__main__":
    main()