代码参考地址:YoloDatasetsEnhance
处理数据增强需要用到images和labels。我保存的labels是txt文件,但是enhance_engine.py中需要对xml文件进行处理,因此需要TxtTransformXml.py将txt文件进行转换。
转换需要修改的地方如下:
路径修改
修改需要转换的xml文件名为“tooth”
目标检测的类别,有几个就填几个,我只有一个类别,就写了“defect”
node_folder.text = 'tooth'
gt_labels = ['defect']
将txt转换为xml文件后,使用数据增强enhance_engine.py,结束后得到xml形式的数据增强labels和数据增强的images。
现在需要将数据增强的xml格式的文件转换为txt文件,使用XmlTransfromTxt.py将xml文件进行转换。
classes = ['defect'] # 标注时的标签 注意顺序一定不要错。