-
Notifications
You must be signed in to change notification settings - Fork 306
Add batch support for VideoDataset and support reading from streams #295
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
tests/test_video_eager.py
Outdated
| video_path = os.path.join( | ||
| os.path.dirname(os.path.abspath(__file__)), "test_video", "small.mp4") | ||
|
|
||
| video_path = "file://" + video_path |
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.
Consider combining with the above two lines?
tests/test_video.py
Outdated
| video_path = os.path.join( | ||
| os.path.dirname(os.path.abspath(__file__)), "test_video", "small.mp4") | ||
|
|
||
| video_path = "file://" + video_path |
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.
Same here
| """A Video File Dataset that reads the video file.""" | ||
|
|
||
| def __init__(self, filename): | ||
| def __init__(self, filename, batch=None): |
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.
Add docstring for this and a note on the relation between dataset's own batch() since I think users would definitely be confused about this.
Signed-off-by: Yong Tang <[email protected]>
Fixes 144 Signed-off-by: Yong Tang <[email protected]>
Signed-off-by: Yong Tang <[email protected]>
|
@terrytangyuan The PR has been updated, please take a look. |
…ensorflow#295) * Add batch support for VideoDataset Signed-off-by: Yong Tang <[email protected]> * Support read video from stream (instead of file name). Fixes 144 Signed-off-by: Yong Tang <[email protected]> * Address review feedback Signed-off-by: Yong Tang <[email protected]>
This fix adds batch support for VideoDataset and support reading from streams.
This fix also fixes #144.
Signed-off-by: Yong Tang [email protected]