Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Default augmentation can translate images but not boxes. #446

Open
Fafa87 opened this issue Mar 4, 2020 · 1 comment
Open

Default augmentation can translate images but not boxes. #446

Fafa87 opened this issue Mar 4, 2020 · 1 comment

Comments

@Fafa87
Copy link

Fafa87 commented Mar 4, 2020

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 #107 or Rodrigo fork https://github.com/rodrigo2019/keras_yolo2 (however mind rodrigo2019#20).

@AIWintermuteAI
Copy link

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)

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants