-
Notifications
You must be signed in to change notification settings - Fork 285
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
Addition of Random frequency mask transform #1680
Addition of Random frequency mask transform #1680
Conversation
Codecov ReportAll modified and coverable lines are covered by tests ✅
Additional details and impacted files@@ Coverage Diff @@
## master #1680 +/- ##
==========================================
+ Coverage 84.74% 84.78% +0.03%
==========================================
Files 153 154 +1
Lines 6465 6480 +15
==========================================
+ Hits 5479 5494 +15
Misses 986 986 ☔ View full report in Codecov by Sentry. |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Thanks a lot for the PR! I left some comments, let me know what you think. The most important is to make k
configurable.
* Replace lambda functions with named functions in the examples
* Move public classes and functions to top in lightly/data * Move public classes and functions to top in lightly/models/modules * Move public classes and functions to top in lightly/utils * Move public classes and functions to top in lightly/loss * Reformat after black failure --------- Co-authored-by: Saurav Maheshkar <[email protected]>
@guarin @philippmwirth I have made the changes, could you guys review the PR now? |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Thanks for the update! There is only one thing left, the default for k should be (0.01, 0.1)
as in the paper.
Done @guarin |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM! 🔥
Modified file(s)
lightly/transforms/__init__.py
: Added new transform importslightly/transforms/random_crop_and_flip_with_grid.py
: Resolvedmypy
issue forAny
subclassingCreated file(s)
lightly/transforms/random_frequency_mask_transform.py
tests/transforms/test_random_frequency_mask_transform.py
Closes: #1660