说明
txt2voc.py程序能够将文件夹中所有label文件转换为xml格式文件,该过程保留label中的所有类别。本程序中,将把上述所有xml文件中bounding box标注的类别筛选成特定的你想要的类别,并把满足条件的图片和txt文件进行保存。
import xml.etree.ElementTree as ET
import pickle
import os
from os import listdir, getcwd
from os.path import join
import shutil
def convert(size, box):
x_center = (box[0]+box[1])/2.0
y_center = (box[2]+box[3])/2.0
x = x_center / size[0]
y = y_center / size[1]
w = (box[1] - box[0]) / size[0]
h = (box[3] - box[