1 安装pycocotools
使用指令:conda/pip install pycocotools
进行安装,如果安装不上,就看这个:
参考:https://blog.csdn.net/qq_51116518/article/details/121207941
2 将YOLO的txt标签格式转换为json格式
这一步参考:YOLO的txt转json以及yolov5使用pycocotools评估
步骤
- 先将YOLO的txt文件生成需要使用的txt文件。
新的生成的txt文件格式:
这个步骤的代码为:
import os
import cv2
'''
function:可将yolo格式的数据集转换为coco格式的(1), 生成annos.txt
需要准备:
labels:yolo格式的标签,是txt格式,名字为图片名
images:原始标签对应的图片,需要有序号
'''
# 原始标签路径E:\pyCharmProject\AI\papercode\datasets\GTSDB
originLabelsDir = '../labels/test'
# 转换后的文件保存路径
saveDir = '../labels/annos1.txt'
# 原始标签对应的图片路径
originImagesDir = '../images/test'
txtFileList = os.listdir(originLabelsDir)
with open(saveDir, 'w') as fw:
for txtFile in txtFileList:
with open(os.path.join(originLabelsDir, txtFile), 'r') as fr:
labelList = fr.readlines()
for label in labelList:
label = label.strip().split()
x = float(label[1])
y