2
$\begingroup$

I don't have a lot of training data and I'm looking for some tools in python or executable program like labelimg that do some heavy augmentation on images, even better if they also change bounding boxes coordinate accordingly. Any help will be appreciated!

$\endgroup$
2
$\begingroup$

I think you should look into imgaug. It supports most image augmentation and does have support for bounding boxes.

Docs: https://imgaug.readthedocs.io/en/latest/

$\endgroup$
  • $\begingroup$ Thank a lot Simon! Just what I'm looking for $\endgroup$ – Hoang Dang Tuan Mar 19 at 6:17
2
$\begingroup$

Other widely used Python libraries for data augmentation include:

  • OpenCV: has functions/methods for bounding boxes, changing color space, scaling, cropping, translation, rotation, filters, blur, thresholding, etc.
  • Scikit-image: also has features allowing converting from another color space to another, resizing/rescaling/rotating, erosion/dilation, filters, etc.
  • Pillow: also has features for color space conversion, resizing/rotation/cropping, filters, pixel-level transformations, etc.
$\endgroup$

Your Answer

By clicking “Post Your Answer”, you agree to our terms of service, privacy policy and cookie policy

Not the answer you're looking for? Browse other questions tagged or ask your own question.