-
Notifications
You must be signed in to change notification settings - Fork 421
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
Refactor _RandomNCrop to RandomCrop + ExtractPatches #2682
Refactor _RandomNCrop to RandomCrop + ExtractPatches #2682
Conversation
0dc0f3f
to
1a025c4
Compare
I would love to get rid of |
2bd3a50
to
b05719f
Compare
I'll bring that back from the grave after this gets merged. I want to see if they can help me with extracting patches but also being able to extract bboxes since this is important for object detection datasets that have large imagery. We don't want to lose users to SAHI. |
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.
Due to the bugs for batch_size=1
and lack of support for bboxes, I'm strongly considering using RandomCrop
for train and CenterCrop
for val/test (with large patch size and small batch size) instead of replacing our hacky _RandomNCrop
with another hacky _ExtractPatches
. Then we can find time to contribute something better upstream to Kornia. How would you feel about that?
I think _ Using CenterCrop this would not be the case and not be able to provide an apples to apples comparison with other baselines. |
91046e4
to
7d8a1a2
Compare
This PR fixes #2667 and refactors
_RandomNCrop
out of segmentation and change detection datamodules in favor ofK.RandomCrop
during training andExtractPatches
for val/test/predicting.