目标检测VOC生成txt文件

VOC数据格式

首先生成ImageSets下的train.txt文件

get_list.py

import os
import random
import shutil

# 设置随机种子
random.seed(1000)

# 判断Annotations和JpegImages是否对应
train_precent = 0.8
label_path = "../../Annotations"
print(os.path.abspath(label_path))
save = "../Main"
print(os.path.abspath(save))
# 所有的标签
total_xml = [os.path.basename(label).split('.')[0] for label in os.listdir(label_path) if label.endswith('.xml')]

# 所有的图片
image_path = '../../JPEGImages'
total_image = [os.path.basename(image).split('.')[0] for image in os.listdir(image_path) if
               image.endswith(('.jpg', '.png', '.bmp'))]

if not os.path.exists("./delete"):
    os.makedirs('./delete')
for label in total_xml:
    # print(os.path.ab

代码007普通

打赏 收藏 海报 链接