自己收集标注整理数据集常见问题

ubantu安装labelImg

注：也可在官网下载labelImg安装包
**下载地址：whl文件
然后在该下载文件目录下执行命令

conda install labelImg-1.8.3-py2.py3-none-any.whl

用pip install会报错说版本不支持2.7的python，因此用pip3

XML文件的问题

4个值有一个是0都不可以，4个都是0，那就是没有标上，更加不可以

 <object>
        <name>face</name>
        <pose>Unspecified</pose>
        <truncated>0</truncated>
        <difficult>0</difficult>
        <bndbox>
            <xmin>186</xmin>
            <ymin>0</ymin>
            <xmax>201</xmax>
            <ymax>14</ymax>
        </bndbox>
    </object>

批量修改label_name

# -*- coding:utf-8 -*-
# Linda Li 2020/5/29 9:08 change_label_name PyCharm
import os
import os.path
from xml.etree.ElementTree import parse, Element


# 批量修改xml中内容 FaceMaskDataset/testxml/
def changeName(xml_fold, origin_name, new_name):
    '''
    xml_fold: xml存放文件夹
    origin_name: 原始名字，比如弄错的名字，原先要cow,不小心打成cwo
    new_name: 需要改成的正确的名字，在上个例子中就是cow
    '''
    files = os.listdir(xml_fold)
    cnt = 0
    for xmlFile in files:
        file_path = os.path.join(xml_fold, xmlFile)
        dom = parse(file_path)
        root = dom.getroot()
        for obj in root.iter('object'): # 获取object节点中的name子节点
            tmp_name = obj.find('name').text
            if tmp_name == origin_name: # 修改
                obj.find('name').text = new_name
                print("change %s to %s." % (origin_name, new_name))
                cnt += 1
        dom.write(file_path, xml_declaration=True)#保存到指定文件
    print("有%d个文件被成功修改。" % cnt)


changeName('FaceMaskDataset/testxml/', 'face_mask', 'mask')