前言
在使用labelimg进行数据标注时可以将结果保存为yolo格式或者voc格式,但是发现要是想用labelimg打开yolo识别后的txt就不行了,在训练ai的时候我们可以用已经有的结果识别一遍图片,然后再进行查漏补缺,这样子效率会更高,这里出一个YOLO的txt与voc的xml相互转换源代码
一、voc转yolo(xml转txt)
import xml.etree.ElementTree as ET
import pickle
import os
from os import listdir, getcwd
from os.path import join
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[