/data_generation_tools

datasets generation for deep learning

Primary LanguagePython

Useful tools for computer vision/deep learning.

简体中文 | English

概要

  • 这是一个针对目标检测场景下的数据增强的项目
  • 本项目特点是针对目标检测数据增强,当生成的图像目标发生形变、偏移、旋转……时,生成的标注文件也会做出相应的变化,能够做到生成后的目标box,仍旧可以严丝合缝地框在目标边缘

数据增强

  • 将图像随机仿射、旋转、平移、拉伸、颜色噪声、亮度等实现数据增强
  • 默认仅针对目标检测数据增强,生成图像及对应的yolo格式标注文件(图像分类需对代码就行部分修改)
  • 你可以指定生成数据的数量 N,但实际生成数据的数量约为 0.8*N

依赖

使用步骤

  • 基于原始图像(data/imgs)和分类标签(data/labels.txt)用在线标注工具 https://www.makesense.ai/ 制作vgg格式的标注文件,也可以导入coco标注文件再导出vgg格式的标注文件 (data/vgg.json 此处文件已存在,用户也可重新制作) image image
  • 生成数据 python generator.py 生成的图像数据存放在(fake/images),对应标签存放在(fake/labels)
  • 验证生成数据 python plot.py 将标注文件的box画到对应的生成图像,画好图像保存在(fake/images)

其他