【目标检测数据集处理】DOTA格式数据集增强「包含图像处理，旋转，缩放等」

最近在做比赛数据预处理的时候，用到了很多数据增强方法，对应的是dota数据集。

数据增强

仓库地址：https://github.com/AO-XIN/dota-augmentation
数据增强包含了下面8种。

使用方法非常简单，将项目clone至本地。

git clone https://github.com/AO-XIN/dota-augmentation.git
pip install -r requirements.txt

使用方法打开main按照注释使用即可。

import aug

A=aug.aug("train/images","/Users/aoxin/CODE/python/dota-augmentation/train/lableTxt","aug-images","aug-lables")# 原始图片路径 原始lables路径 扩充到的图片路径 扩充到的lables路径
# A.Rotate(angle=90)

# A.RandomResize()
A.AddWeather()

其他工具

项目中还有一些其他小工具，比如compare.py和vision.py

• vision.py是对dota数据集进行可视化将框标注在图片上。
• compare.py是将对图片文件夹和标签文件夹进行检查，查看是否有标签及图片遗漏。