当前位置:   article > 正文

66、将同图片下的多个不同类别的xml标注文件合并成一个xml标注文件-labelImg格式

66、将同图片下的多个不同类别的xml标注文件合并成一个xml标注文件-labelImg格式

基本思想:手中有一套抽烟的数据集是labelimg格式,但是没有人物标注的数据集,因此使用自动化标注脚本将图片过滤一边,进行生成labelimg文件,只含有80类别的人物标注xml,然后使用脚本将生成标注的人物xml和手中有的抽烟xml进行合并,生成一份xml文件

代码

  1. # -*- coding: utf-8 -*-
  2. import glob
  3. import xml.etree.ElementTree as ET
  4. import os
  5. from xml.etree.ElementTree import parse, Element
  6. path_0 = '/home/ubuntu/Downloads/smoke-used/smoke/images/xml_file_0' #来自自动化生成人物xml标注文件
  7. path_1 = '/home/ubuntu/Downloads/smoke-used/smoke/images/xml_file_1' #来自公开数据集抽烟的xml标注文件
  8. destPath = "/home/ubuntu/Downloads/smoke-used/smoke/images/combine" #合并的xml标注文件 格式lableImg
  9. xml_0=glob.glob(os.path.join(path_0,"*.xml"))
  10. xml_1=glob.glob(os.path.join(path_1,"*.xml"))
  11. for item_0 in xml_0:
  12. name_0=os.path.split(item_0)
  13. for item_1 in xml_1:
  14. name_1 = os.path.split(item_1)
  15. new_location=os.path.join(destPath,name_1[-1])
  16. if name_1[-1]==name_0[-1]:
声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/我家自动化/article/detail/280587
推荐阅读