-
Notifications
You must be signed in to change notification settings - Fork 618
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
Fix missing source info in the numpy reader output #4714
Conversation
!build |
CI MESSAGE: [7594045]: BUILD STARTED |
CI MESSAGE: [7594045]: BUILD FAILED |
CI MESSAGE: [7594045]: BUILD PASSED |
@@ -658,13 +669,10 @@ def check_pad_last_sample(device): | |||
for _ in range(2): | |||
pipe_out = pipe.run() | |||
for i in range(batch_size): | |||
pipe_arr = to_array(pipe_out[0][i]) | |||
out_arr = to_array(pipe_out[0][i]) | |||
out_prop = to_array(pipe_out[1][i]) |
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.
SourceInfo has a first-class support, so you could just do:
out_prop = to_array(pipe_out[1][i]) | |
out_prop = pipe_out[0][i].source_info() |
This way there wouldn't be a need to use get_property()
in the pipeline.
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.
Fixed
- fixes missing copy of the source info metadata from samples read in the numpy reader (CPU and GPU) to the operator output Signed-off-by: Janusz Lisiecki <jlisiecki@nvidia.com>
22502ee
to
c5eb015
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.
With optional improvements.
Signed-off-by: Janusz Lisiecki <jlisiecki@nvidia.com>
!build |
CI MESSAGE: [7673596]: BUILD STARTED |
CI MESSAGE: [7673596]: BUILD FAILED |
CI MESSAGE: [7673596]: BUILD PASSED |
- fixes missing copy of the source info metadata from samples read in the numpy reader (CPU and GPU) to the operator output Signed-off-by: Janusz Lisiecki <jlisiecki@nvidia.com>
samples read in the numpy reader (CPU and GPU) to the
operator output
Category:
Other (e.g. Documentation, Tests, Configuration)
Description:
samples read in the numpy reader (CPU and GPU) to the
operator output
Additional information:
Affected modules and functionalities:
Key points relevant for the review:
Tests:
Checklist
Documentation
DALI team only
Requirements
REQ IDs: N/A
JIRA TASK: N/A