文章内容
将labelme的内容转换成yolo需要的txt文件

python">import json
import os
import shutil
import random# Convert label to idx
# with open("labels.txt", "r") as f:
#     classes = [c.strip() for c in f.readlines()]
#     idx_dict = {c: str(i) for i, c in enumerate(classes)}# 传入3个路径
# 获取名称，通过名称和路径来确定生成的地址
def onefile_labelme_to_yolo(json_path,txt_path,img_path):# 检查输入的路径是否为有效的文件夹路径if os.path.isdir(json_path):# 遍历文件夹中的所有文件和文件夹for root, dirs, files in os.walk(json_path):for file in files:# 分离文件名和扩展名file_name, file_extension = os.path.splitext(file)# 生成xml文件路径txt_file = os.path.join(txt_path, file_name + '.txt')json_file = os.path.join(json_path, file)print(txt_file)# 读取json文件with open(json_file, 'r') as f:labelme_data = json.load(f)# 解析json文件image_filename = labelme_data["imagePath"]image_width = labelme_data["imageWidth"]image_height = labelme_data["imageHeight"]with open(txt_file, 'w') as f:for shape in labelme_data["shapes"]:label = shape["label"]points = shape["points"]x_min, y_min = points[0]x_max, y_max = points[1]center_x = round(((x_min + x_max) / 2) / image_width, 2)center_y = round(((y_min + y_max) / 2) / image_height, 2)width = round(abs(x_min - x_max) / image_width, 2)height = round(abs(y_min - y_max) / image_height, 2)# class_id = label_dict[label]f.write(f"{0}  {center_x:.6f} {center_y:.6f} {width} {height}\n")print(f"Converted {file} to {txt_path}")else:print(f"{json_path} 不是一个有效的文件夹路径。")if __name__ == '__main__':labelme_folder = "G:\\2025_study\\json"  # labelme生成的标注文件所在的文件夹txt_folder = "G:\\2025_study\\txt"  # 存储yolo标注文件的文件夹img_folder = "G:\\2025_study\\img"onefile_labelme_to_yolo(labelme_folder,txt_folder,img_folder)