Advertisement

数据集转换为COCO格式为txt格式

阅读量:

json_to_txt

  • 文本格式
    • 环境配置
    • 程序代码
    • 转换后的路径
    • 实验结果

txt格式

tensorflow版的yolov3进行训练所需的数据格式为txt文件,具体要求如下:

xxx/xxx.jpg 18.19,6.32,424.13,421.83,20 323.86,2.65,640.0,421.94,20
xxx/xxx.jpg 48,240,195,371,11 8,12,352,498,14
image_path x_min, y_min, x_max, y_max, class_id x_min, y_min ,…, class_id
应确保x_max小于图像宽度且y_max小于图像高度

关于tensorflow-yolov3模型,其class_id的编号方式是从0开始递增至类别总数减一。然而,在我所使用的数据集中,category_id的数值分别为1、2、3、4、8、9、10。因此,我在代码中对boundingBox所属类别部分进行了相应的调整与补充。

复制代码
    	id = boundingBox_per['category_id']    

全部评论 (0)

还没有任何评论哟~