-
Notifications
You must be signed in to change notification settings - Fork 705
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
padim arguments improvements #664
padim arguments improvements #664
Conversation
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 @jpcbertoldo. The timing of this PR is also great, given that we wanted to add Efficient backbone support.
Cool! I will write some the doc for it. |
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.
Looks good to me
ffb2190
to
9735cd8
Compare
done |
…jpcbertoldo/anomalib into jpcbertoldo/padim-config-improvements-2
@djdameln it didn't feel right to have such a specific test. Instead i extracted the dryrun part and tested that. The logic in |
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.
There is going to be merge conflicts between this PR and #675. @ashwinvaidya17 and @jpcbertoldo, can you guys sync how to minimise/avoid these conflicts?
The conflict is only in Doesnt look so hard, whicever is merged second rebases. @ashwinvaidya17 could you check if the new function in that file in #675 would work with the torchfx extractor? If not then it needs a subcase. |
I am working on a design to make the feature extractor configurable. I think we can move dimension computation into the feature extractor class. Since the design might take time we can merge this one first and then I can update PR #675 |
Cool : ) |
Description
Fix #662 by creating the parameter
n_features
(nb of features/axes retained by the dimension reduction).I also made it possible to use a custom backbone without modifying the source code.
This was limited by this hardcoded dict
If the backbone was not in the dict then the
orig_dims
andemb_scale
could not be set.These two don't need to be hardcoded, I made it be deduced from a dryrun feature extraction.
Changes
Checklist