YOLO系列小样本数据集进行数据增强

代码参考地址: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']  # 标注时的标签 注意顺序一定不要错。

最近更新

  1. docker php8.1+nginx base 镜像 dockerfile 配置

    2024-04-13 10:36:03       98 阅读
  2. Could not load dynamic library ‘cudart64_100.dll‘

    2024-04-13 10:36:03       106 阅读
  3. 在Django里面运行非项目文件

    2024-04-13 10:36:03       87 阅读
  4. Python语言-面向对象

    2024-04-13 10:36:03       97 阅读

热门阅读

  1. 嵌入式之数据结构篇(五)

    2024-04-13 10:36:03       38 阅读
  2. Gobalt Strike基础

    2024-04-13 10:36:03       32 阅读
  3. docker简介和运行命令

    2024-04-13 10:36:03       40 阅读
  4. Docker一键安装Memos个人笔记

    2024-04-13 10:36:03       31 阅读