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

Added scale augmentation #236

Open
wants to merge 3 commits into
base: main
Choose a base branch
from

Conversation

hummuscience
Copy link
Contributor

So, I did some testing and adding the scale augmentation improved all the models that include images from different sources (where the animals have different sizes).

I will post a more quantitative analysis when everything is done, but visually things are better and the switches are gone!

I implemented two options. Scaling + DLC and scaling + DLC + horizontal/vertical flips.

It is getting a bit wild with all the options. I wondered if it makes sense to let the user pass multiple augmentation building blocks and then build the logic around that. So the user passes a list like so: [dlc, topdown, lr, scale].

I also noticed that currently the exact settings of the augmentation can't be changed in the config. Maybe it's a bit overkill for now, but I could imagine augmentations being its own part of the config in the future.

@themattinthehatt
Copy link
Collaborator

themattinthehatt commented Jan 20, 2025

cool @hummuscience, thanks for testing this out. I'm excited to see some of your qualitative results!

I agree that the options are getting a bit out of hand. let me talk with @ksikka and figure out what a good long-term solution is (I like your idea of a list). We can also figure out how to allow users to change augmentation parameters easily. Previously we had hidden all these options so as not to overwhelm people, but it makes sense that more advanced users will want control over these.

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

Successfully merging this pull request may close these issues.

2 participants