赞
踩
基本思想:手中有一套抽烟的数据集是labelimg格式,但是没有人物标注的数据集,因此使用自动化标注脚本将图片过滤一边,进行生成labelimg文件,只含有80类别的人物标注xml,然后使用脚本将生成标注的人物xml和手中有的抽烟xml进行合并,生成一份xml文件
代码
- # -*- coding: utf-8 -*-
- import glob
-
- import xml.etree.ElementTree as ET
- import os
- from xml.etree.ElementTree import parse, Element
-
- path_0 = '/home/ubuntu/Downloads/smoke-used/smoke/images/xml_file_0' #来自自动化生成人物xml标注文件
- path_1 = '/home/ubuntu/Downloads/smoke-used/smoke/images/xml_file_1' #来自公开数据集抽烟的xml标注文件
- destPath = "/home/ubuntu/Downloads/smoke-used/smoke/images/combine" #合并的xml标注文件 格式lableImg
-
-
-
- xml_0=glob.glob(os.path.join(path_0,"*.xml"))
- xml_1=glob.glob(os.path.join(path_1,"*.xml"))
-
- for item_0 in xml_0:
- name_0=os.path.split(item_0)
- for item_1 in xml_1:
- name_1 = os.path.split(item_1)
- new_location=os.path.join(destPath,name_1[-1])
- if name_1[-1]==name_0[-1]:
-
