Quick implemention for data augmentation via imgaug
,related usage refer to rotate-yolo.
(Or you can run the demo I leave behind the code,it's easy to replicate the effect.)
This code is completed for DOTA dataset
, thus specific modification is necessary for your own label.
- Supported methods :affine(sheer、translation、scale、rotation)、hsv augmentation、noise、flip、blur
- You're supposed to run one method everytime , and each aug_functioin wil output img files as well as corresponding xml_label files .
- Customed: you need to rewrite
read_xml
andrewrite_label
function for correct label process.
- HSV
- Affine
- Flip
- Noise
- Blur