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

update detection transformation to allow tensor input #1111

Merged
merged 1 commit into from
Jul 12, 2019

Conversation

buoyancy99
Copy link
Contributor

@buoyancy99 buoyancy99 commented Jul 12, 2019

Though official document says the rcnns accepts a list of images. But it also makes sense to accept a batched tensor directly. The change eliminates bugs triggered when passing in tensor

@codecov-io
Copy link

codecov-io commented Jul 12, 2019

Codecov Report

Merging #1111 into master will decrease coverage by 0.05%.
The diff coverage is 100%.

Impacted file tree graph

@@            Coverage Diff             @@
##           master    #1111      +/-   ##
==========================================
- Coverage   64.85%   64.79%   -0.06%     
==========================================
  Files          68       68              
  Lines        5417     5417              
  Branches      834      835       +1     
==========================================
- Hits         3513     3510       -3     
- Misses       1647     1648       +1     
- Partials      257      259       +2
Impacted Files Coverage Δ
torchvision/models/detection/transform.py 78.3% <100%> (ø) ⬆️
torchvision/transforms/transforms.py 80.35% <0%> (-0.59%) ⬇️

Continue to review full report at Codecov.

Legend - Click here to learn more
Δ = absolute <relative> (impact), ø = not affected, ? = missing data
Powered by Codecov. Last update dea1afb...ebc3905. Read the comment docs.

Copy link
Member

@fmassa fmassa left a comment

Choose a reason for hiding this comment

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

Thanks!

@fmassa fmassa merged commit 8faaefc into pytorch:master Jul 12, 2019
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants