欢迎来到尧图网

客户服务 关于我们

您的位置:首页 > 文旅 > 八卦 > 目标检测-AnyLabeling标注格式转换成YOLO格式

目标检测-AnyLabeling标注格式转换成YOLO格式

2024/11/30 8:51:38 来源:https://blog.csdn.net/weixin_49824703/article/details/139310197  浏览:    关键词:目标检测-AnyLabeling标注格式转换成YOLO格式

Anylabel可以极大的增加数据的标注效率,但是其标注格式如何能转换成YOLO标注格式,具体内容如下所示。


关于AnyLabeling的其它详细介绍如下链接所示

https://blog.csdn.net/u011775793/article/details/134918861

 Github链接

https://github.com/vietanhdev/anylabeling

python代码


import json
import osdef labelme_to_yolo(label_me_json_file, cls2id_dict):label_me_json = json.load(open(label_me_json_file, mode='r', encoding='UTF-8'))shapes = label_me_json['shapes']img_width, img_height = label_me_json['imageWidth'], label_me_json['imageHeight']img_path = label_me_json['imagePath']img_data = label_me_json['imageData'] if 'imageData' in label_me_json else ''labels = []for s in shapes:s_type = s['shape_type']s_type = s_type.lower()if s_type == 'rectangle':pts = s['points']x1, y1 = pts[0]  # left cornerx2, y2 = pts[1]  # right cornerx = (x1 + x2) / 2 / img_widthy = (y1 + y2) / 2 / img_heightw = abs(x2 - x1) / img_widthh = abs(y2 - y1) / img_heightcid = cls2id_dict[s['label']]labels.append(f'{cid} {x} {y} {w} {h}')return labelsdef write_label2txt(save_txt_path,label_list):f=open(save_txt_path,"w",encoding="UTF-8")for label in label_list:temp_list=label.split(" ")f.write(temp_list[0])f.write(" ")f.write(temp_list[1])f.write(" ")f.write(temp_list[2])f.write(" ")f.write(temp_list[3])f.write(" ")f.write(temp_list[4])f.write("\n")if __name__ == '__main__':# 原始图片文件夹路径img_dir=r"D:\desk\Work\Dataset\Test\Test_Anylabeling\imgs"# 原始JSON标签文件夹路径json_dir=r"D:\desk\Work\Dataset\Test\Test_Anylabeling\labels"# 生成保存TXT文件夹路径save_dir=r"D:\desk\Work\Dataset\Test\Test_Anylabeling\txt"# 类别和序号的映射字典cls2id_dict={"building1":"0"}if not os.path.exists(save_dir):os.makedirs(save_dir)for json_name in os.listdir(json_dir):json_path=os.path.join(json_dir,json_name)txt_name=json_name.split(".")[0]+".txt"save_txt_path=os.path.join(save_dir,txt_name)labels = labelme_to_yolo(json_path,cls2id_dict)write_label2txt(save_txt_path,labels)

具体修改 

# 原始图片文件夹路径
img_dir=r"D:\desk\Work\Dataset\Test\Test_Anylabeling\imgs"

改为自己的图片路径

 

# 原始JSON标签文件夹路径
json_dir=r"D:\desk\Work\Dataset\Test\Test_Anylabeling\labels"

改为自己的JSON文件夹路径

# 生成保存TXT文件夹路径
save_dir=r"D:\desk\Work\Dataset\Test\Test_Anylabeling\txt"

改为自己的保存生成的yolo文件夹路径

 

# 类别和序号的映射字典
cls2id_dict={"building1":"0"}

改为自己的标签映射

 

开始实验

实验准备

 

运行代码

实验验证

实验验证可视化代码相关链接

https://blog.csdn.net/weixin_49824703/article/details/134050547

完美!

版权声明:

本网仅为发布的内容提供存储空间,不对发表、转载的内容提供任何形式的保证。凡本网注明“来源:XXX网络”的作品,均转载自其它媒体,著作权归作者所有,商业转载请联系作者获得授权,非商业转载请注明出处。

我们尊重并感谢每一位作者,均已注明文章来源和作者。如因作品内容、版权或其它问题,请及时与我们联系,联系邮箱:809451989@qq.com,投稿邮箱:809451989@qq.com