-
Notifications
You must be signed in to change notification settings - Fork 621
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
Improve support for labels in VideoReader. #1500
Improve support for labels in VideoReader. #1500
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.
Can you add some test for this?
Maybe some generation of file_list on the fly inside of the test and verification if we get the expected lables would suffice.
Signed-off-by: Abhishek Sansanwal <asansanwal@nvidia.com>
Signed-off-by: Abhishek Sansanwal <asansanwal@nvidia.com>
db906b0
to
018e096
Compare
CI MESSAGE: [1015170]: BUILD STARTED |
018e096
to
bae68fb
Compare
bae68fb
to
b8920a7
Compare
CI MESSAGE: [1020646]: BUILD STARTED |
Description: Added gtest cases for enable_frame_num, enable_timestamps, file_list_frame_num and labels to parts of a video. Signed-off-by: Abhishek Sansanwal <asansanwal@nvidia.com>
b8920a7
to
f9d31a2
Compare
CI MESSAGE: [1020778]: BUILD STARTED |
CI MESSAGE: [1020823]: BUILD STARTED |
CI MESSAGE: [1020823]: BUILD FAILED |
CI MESSAGE: [1020823]: BUILD PASSED |
CI MESSAGE: [1020823]: BUILD FAILED |
CI MESSAGE: [1020823]: BUILD PASSED |
f9d31a2
to
af864d7
Compare
CI MESSAGE: [1024370]: BUILD STARTED |
CI MESSAGE: [1024370]: BUILD FAILED |
CI MESSAGE: [1024549]: BUILD STARTED |
CI MESSAGE: [1024549]: BUILD PASSED |
Signed-off-by: Abhishek Sansanwal <asansanwal@nvidia.com>
18b9d34
to
89f4b6f
Compare
Signed-off-by: Abhishek Sansanwal asansanwal@nvidia.com
Why we need this PR?
Pick one
Add's support for applying labels to part parts of a video. Either using timestmap or frame number through file_list.
Add's support for outputting frame_numbers and timestamps of the decoded frames using
enable_frame_num
andenable_timestamps
parametersAddresses #940, #1458, #753, #666
What happened in this PR?
Explain solution of the problem, new feature added.
Two additional columns can be provided as input using the plaintxt file passed through file_list argument.
The two values are interpreted as start/end timestamps. If they are skipped the label is applied to the entire video. If
file_list_frame_num
is set to true, then the timestamps are interpreted as frame numbers instead of timestamps.What was changed, added, removed?
Parsing the input file_list was modified.
Logic to generate sequences was changed to apply labels to sequences that fall within the start/end range specified.
What is most important part that reviewers should focus on?
filesystem::get_file_label_pair() in video_loader.cc
PrepareMetadataImpl() in video_loader.h
Was this PR tested? How?
Tested using gtest test app.Using
gtest_filter=VideoReaderTest.StartEndLabels:VideoReaderTest.FrameLabels:VideoReaderTest.FrameLabelsWithFileListFrameNum:VideoReaderTest.TimestampLabels
Were docs and examples updated, if necessary?
Docs updated.
Needs NVIDIA/DALI_extra#35 for test cases
Future to-do ?
Support reading input from csv files instead of plaintext files.
JIRA TASK: [DALI-XXXX]