MakeSense选择并保存数据集的图片

---

前言

根据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

---

总结

以上就是今天要讲的内容