从VOC数据集中提取部分图片对应的标签文件

从VOC数据集中提取部分图片对应的标签文件

import shutil
import os

def main():
    # 原始VOC标签存放目录
    path_label = './VOCdevkit/VOC2007/Annotations'
    # 部分图片
    path_object = './mydata/VOC2007/JPEGImages_bus'
    type_object = '.xml'
    # 从原始标签中提取部分图片对应标签的保存目录
    path_output = './mydata/VOC2007/Annotations_ssd'

    # 遍历标签和图片文件夹
    imagelist = os.listdir(path_object)
    labellist = os.listdir(path_label)

    for x in imagelist:
        temp = x.split(".")[0] + type_object
        obj_path = os.path.join(path_label, temp)
        new_path = os.path.join(path_output, temp)
        # 如果部分图片对应的标签名称 在 原始标签文件夹中 则将其提取
        if temp in labellist:
            print(temp)
            shutil.copyfile(obj_path, new_path)

if __name__ == '__main__':
    main()