-
Notifications
You must be signed in to change notification settings - Fork 7k
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
extend video reader to support fast video probing #1437
extend video reader to support fast video probing #1437
Conversation
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.
Thanks for the PR!
This is ok for now, but I think we should probably refactor in the future the way we expose those attributes properties.
Also, C++ linter is failing in https://travis-ci.org/pytorch/vision/jobs/595456375, could you fix it?
The osx_wheel failures are unrelated to this PR.
Once linter is fixed this is good to merge.
Codecov Report
@@ Coverage Diff @@
## master #1437 +/- ##
==========================================
+ Coverage 64.01% 64.11% +0.09%
==========================================
Files 80 80
Lines 6308 6325 +17
Branches 968 970 +2
==========================================
+ Hits 4038 4055 +17
+ Misses 1989 1984 -5
- Partials 281 286 +5
Continue to review full report at Codecov.
|
@fmassa Make sense. Let's do it in the next refactoring PR. |
f5cf884
to
f4708cf
Compare
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.
Thanks a lot Zhicheng!
Summary: Pull Request resolved: #62 Current dependency torchvision 0.4.0 was released in August. It missed quite a few PRs that are merged after that, and that are needed for video classification, such as - pytorch/vision#1437 - pytorch/vision#1431 - pytorch/vision#1423 - pytorch/vision#1418 - pytorch/vision#1408 - pytorch/vision#1376 - pytorch/vision#1363 - pytorch/vision#1353 - pytorch/vision#1303 This will fail the CI test when a diff uses changes made in those PRs. Before a new official version of TorchVision is released, we can temporarily use the nightly torchvision to get all the recent PRs, and unblock the PR merging. We plan to use a fixed version of TorchVision later. Reviewed By: vreis Differential Revision: D17944239 fbshipit-source-id: 86ff540e3fc4f08ef767e84ef103525db5158201
* extend video reader to support fast video probing * fix c++ lint * small fix * allow to accept input video of type torch.Tensor
Changes
Extend video reader to support probing the video quickly for getting video meta information
io
module, add new methods_probe_video_from_file
and_probe_video_from_memory
unit test
python test/test_video_reader.py
python test/test_io.py