keras-yolo2 icon indicating copy to clipboard operation
keras-yolo2 copied to clipboard

Default augmentation can translate images but not boxes.

Open Fafa87 opened this issue 4 years ago • 1 comments

sometimes(iaa.Affine(...)) is used in default augmentation in master which can rotate and translating the image. Boxes stay on the same place with is bad (especially if you have small boxes).

The fix is to use imaug to move bboxes as well. See https://github.com/experiencor/keras-yolo2/issues/107 or Rodrigo fork https://github.com/rodrigo2019/keras_yolo2 (however mind https://github.com/rodrigo2019/keras_yolo2/issues/20).

Fafa87 avatar Mar 04 '20 10:03 Fafa87

Hi! I have noticed this problem as well. Augmentation has been fixed in https://github.com/experiencor/keras-yolo3 or alternatively in my project https://github.com/AIWintermuteAI/aXeleRate which also supports other network types(classifier and segnet). You can check dataset augmentation in colab notebook https://colab.research.google.com/github/AIWintermuteAI/aXeleRate/blob/master/resources/aXeleRate_test_detector.ipynb by executing visualize_dataset(img_folder='aXeleRate/sample_datasets/detector/imgs', ann_folder='aXeleRate/sample_datasets/detector/anns', img_size=None, jitter=True)

AIWintermuteAI avatar Mar 17 '20 06:03 AIWintermuteAI