本项目的功能是对YOLO格式的数据实现数据增强,使用本项目代码可以较快的完成扩充数据集快速扩充的工作,会对图像和标注同时处理。但其也存在一定的问题,例如无法保证处理后的数据一定能跑出更好的效果,处理结束后可能会出现一些损坏的图片(原图过大时)。
但本项目处理后出现的坏图,在YOLOv5中会被识别出来并不做训练,其实对训练影响不大,追求完美的同学可以考虑不使用resize系列的函数,手动剔除坏图,使用其他数据增强项目完成工作等。
使用
1. download本项目到本地(建议下载到有图形界面的操作系统中)
2. 打开DataAugOnDetectin.py
修改 image_path, label_path, save_path 三个参数
```
image_path = "" # 图片的路径
label_path = "" # 标签文件的路径
save_path = "" # 数据增强的结果保存位置路径
```
3. 运行,使用pycharm或spyder等软件运行DataAugOnDetectin.py
2024-04-02 13:07:26
897KB
python
1