前言
根据make sense得到的标准文件,选择并保存数据集的图片
代码写的很菜仅供参考
一、目标检测数据集标注
有时候labelImg会闪退,我一般选择用MakeSense标注图片,而且可以选择生成VOC格式还是YOLO格式,这段代码用于根据make sense得到的标准文件,选择并保存数据集的图片
二、使用步骤
1.引入库
代码如下(示例):
import os
import cv2
2.读入数据
代码如下(示例):
print(len(os.listdir(r'D:\pycharm\project\yolov5-pytorch\VOCdevkit\VOC2016\Annotations')))#makesense得到的解压后的标注文件的地址
for name in os.listdir(r'D:\pycharm\project\yolov5-pytorch\VOCdevkit\VOC2016\Annotations'):
imgname=os.path.splitext(name)[0]+'.jpg'
img=cv2.imread(os.path.join('D:/pycharm/tongue',imgname))#图片总的地址
#print(img)
cv2.imwrite(os.path.join('D:/pycharm/project/yolov5-pytorch/VOCdevkit/VOC2016/JPEGImages',imgname), img)#保存后的地址
print('{}over'.format(name))
3.用于判断xml文件夹与image文件夹的数量是否相同
lst1=os.listdir(r'D:\pycharm\acne04\Detection\VOC2007\JPEGImages')
lst2=os.listdir(r'D:\pycharm\acne04\Detection\Annotations')
i=0
for a in lst1:
if os.path.splitext(a)[0] == os.path.splitext(lst2[i])[0]:
continue
print('{}isover'.format(os.path.splitext(a)[0]))
i+=1
总结
以上就是今天要讲的内容