手把手教你制作Pytorch-YOLOv4训练集 xml转txt 附代码附注释

最近想用github上面Tianxiaomo写的pytorch-YOLOv4版本代码跑自己的训练集,网上没找到现成的制作txt训练集代码,所以对着别人的改了一个
参考博客:将xml转换成txt格式
匹配github上面Tianxiaomo的pytorch-YOLOv4代码,可以直接制作和转换对应的训练集

准备

labelImg制作自己的训练集,格式为xml,上面标注了图片上锚框的各种信息
图像标注软件——labelImg使用教程

xml内容

训练集格式

修改代码

class_names改成你自己定义的类别,比如我这里是dog和cat
xml_path 和 txt_path 改成你笔记本xml文件和将要保存的txt文件路径

效果

代码

import os
import os.path
import glob
# xml为自带库
import xml.etree.ElementTree as ET


# 训练集信息
class_names = ['', '']	# 训练集标签, 比如 dog, cat 

# 获取文件路径
# xm