-
Notifications
You must be signed in to change notification settings - Fork 7.1k
Promote prototype transforms to beta status #7261
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
Conversation
…nto migration_transforms_v2
…vision into migration_transforms_v2
@@ -1089,12 +1089,18 @@ def make_datapoints(self, with_mask=True): | |||
"t_ref, t, data_kwargs", | |||
[ | |||
(det_transforms.RandomHorizontalFlip(p=1.0), prototype_transforms.RandomHorizontalFlip(p=1.0), {}), | |||
(det_transforms.RandomIoUCrop(), prototype_transforms.RandomIoUCrop(), {"with_mask": False}), | |||
# FIXME: make |
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.
@pmeier what's broken here ?
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.
After removing the Label
and OneHotLabel
classes, RandomIoUCrop
no longer removes labels since it doesn't have a notion of them. To resolve this, we need to follow my suggestion in the comment and use a composed transform with RandomIoUCrop
and the new SanitizeBoundingBoxes
that @NicolasHug implemented.
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.
prototype tests are healthy, as discussed offline the one remaining thing is to remove those APIs from prototype
that we have migrated to torchvision
. I'll pre-approve now so that you can merge as soon as this is done and not be blocked
Failed tests are related: https://github.com/pytorch/vision/actions/runs/4188549423/jobs/7259827006 ? |
Yes. Will fix them. |
Relevant CI is green. Let's do this 🎉 |
Summary: Reviewed By: vmoens Differential Revision: D44416605 fbshipit-source-id: 0fd313c8279f6be9ed488a19e2d053f33c2f77a5 Co-authored-by: Nicolas Hug <[email protected]> Co-authored-by: vfdev-5 <[email protected]>
cc @vfdev-5 @bjuncek