-
Notifications
You must be signed in to change notification settings - Fork 7.1k
port image type conversion transforms to prototype API #5640
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
💊 CI failures summary and remediationsAs of commit ec1622c (more details on the Dr. CI page): 💚 💚 Looks good so far! There are no failures yet. 💚 💚 This comment was automatically generated by Dr. CI (expand for details).Please report bugs/suggestions to the (internal) Dr. CI Users group. |
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.
The overarching question is: do we want to have a ConvertImageType
transform that takes all kind of image types and converts them to a fixed format? That could unify all three of the transforms below. As output formats, I would only allow tensor and PIL images for now.
torchvision/prototype/transforms/functional/_type_conversion.py
Outdated
Show resolved
Hide resolved
Conflicts: torchvision/prototype/transforms/__init__.py
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.
Overall looks good. Just a few questions:
Conflicts: torchvision/prototype/transforms/__init__.py
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, thanks!
Summary: * port image type conversion transforms to prototype API * implement proposal for image type conversion * add deprecation warnings * appease mypy Reviewed By: NicolasHug Differential Revision: D35393162 fbshipit-source-id: e1be4ecfc2c018d24431e66da7d98d32967944a1 Co-authored-by: Vasilis Vryniotis <[email protected]>
This is a 1-to-1 port of the old transformations. This should serve as discussion basis.