0
点赞
收藏
分享

微信扫一扫

Kafka Consumer 消费消息和 Rebalance 机制

柠檬的那个酸_2333 2024-03-25 阅读 9
python

基本思想:手中有一套抽烟的数据集是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]:

            
举报

相关推荐

0 条评论