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

About data augmentation in [tiny] yolo #6130

Open
spinlud opened this issue Jul 4, 2020 · 4 comments
Open

About data augmentation in [tiny] yolo #6130

spinlud opened this issue Jul 4, 2020 · 4 comments

Comments

@spinlud
Copy link

spinlud commented Jul 4, 2020

Hi guys!
I am using yolov3-tiny and yolov4-tiny to train a custom model. For my current task I don't need any rotation, horizontal/vertical flip and skewness on the training images. Do I need to change anything in the .cfg file to disable these types of data augmentation or are they already disabled?

Looking for example inside the yolov4-tiny.cfg file I see angle=0, does this mean that rotation augmentation is disabled? And what about horizontal/vertical flip and skewness, do I need to do anything? Is there a list of the parameters used in yolo with a description about what they do?

Thanks for any info!

@AlexeyAB
Copy link
Owner

AlexeyAB commented Jul 4, 2020

@spinlud
Copy link
Author

spinlud commented Jul 4, 2020

Thank you @AlexeyAB, exactly what I was looking for!
Hope you don't mind, I have some doubts about the following:

  1. scale_x_y, resize and jitter, what is the difference?
  2. angle=0 means that no rotation augmentation is added, is this right?
  3. adversarial_lr=1.0 this is very fascinating and mysterious, could you elaborate a bit what does it mean Changes all detected objects to make it unlike themselves from neural network point of view? In practice what happens?
  4. All the augmentation parameters (hue, saturation, mosaic, blur, etc) must be placed in the [net] layer, correct?

@sealedtx
Copy link

@spinlud
Here is how I understand these params:

  1. scale_x_y is used to eliminate grid sensitivity, you can read more here Sensitivity Effects Near Grid Boundaries (Experimental Results) ~+1 AP@[.5, .95] #3293
    resize - randomly resizes image in range: 1/1.5 - 1.5 with KEEPING aspect ratio of image
    jitter - randomly crops and resizes images with CHANGING aspect ratio of image
  2. yes, 0 - means no rotation added
  3. adversarial_lr - makes model more stable to adversarial-attacks, read more here Self-adversarial training - data augmentation #5117
  4. correct, read more wiki

@spinlud
Copy link
Author

spinlud commented Jul 14, 2020

Thanks a lot!

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

3 participants