Skip to content
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

Remove torchvision.io from test_utils.py #3092

Merged
merged 1 commit into from
Dec 2, 2020

Conversation

fmassa
Copy link
Member

@fmassa fmassa commented Dec 2, 2020

For now this simplifies a few things for fbcode integration. We can add this back in the future

For now this simplifies a few things for fbcode integration. We can add this back in the future
@fmassa fmassa requested a review from datumbox December 2, 2020 10:10
@fmassa fmassa merged commit a0e5bbe into pytorch:master Dec 2, 2020
@fmassa fmassa deleted the remove_io_from_test_utils branch December 2, 2020 10:59
@@ -90,9 +90,10 @@ def test_draw_boxes(self):

path = os.path.join(os.path.dirname(os.path.abspath(__file__)), "assets", "fakedata", "draw_boxes_util.png")
if not os.path.exists(path):
write_png(result, path)
res = Image.fromarray(result.permute(1, 2, 0).contiguous().numpy())
res.save(path)
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

This was recently changed based on your comment. What's the issue using the io.image here?

Copy link
Member Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Right, my bad sorry about that.

Internal fb tests were failing because this test now imposes that the image part of torchvision is compiled, and we would need to adapt some TARGETS file for that. In order to move forward with fbcode integration, it's easier to first change to using PIL, then update the TARGETS file in fbcode and then add back the dependency to torchvision.io.

vfdev-5 pushed a commit to Quansight/vision that referenced this pull request Dec 4, 2020
For now this simplifies a few things for fbcode integration. We can add this back in the future
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants