最近在训练自己的yolo模型,训练之后想算mAP,发现网络上基本都是VOC数据集的标签制作方法。
我的标签一开始是这样的:
![b9c9571d7699e4cf3217fa99b0bb1654.png](https://i-blog.csdnimg.cn/blog_migrate/92bacf61c09bc97a955fa6d7fec007a8.jpeg)
所以和VOC的不一样,于是就自己做xml文件,附代码:
from xml.dom.minidom import Document
import os
import cv2
def makexml(txtPath,xmlPath,picPath): #读取txt路径,xml保存路径,数据集图片所在路径
dict = {
'0': "slit_l",#字典对类型进行转换
'1': "trans_l",
'2': "slot_l",
'3': "sink_l",
'4': "chap_l",
'5': "block_l",
'6': "track_l",
'7': "