-
Notifications
You must be signed in to change notification settings - Fork 6.9k
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
test output_format in video datasets #7879
Conversation
|
||
def __init__(self, *args, **kwargs): | ||
super().__init__(*args, **kwargs) | ||
self.dataset_args = self._set_default_frames_per_clip(self.dataset_args) | ||
|
||
def _set_default_frames_per_clip(self, inject_fake_data): | ||
def _set_default_frames_per_clip(self, dataset_args): |
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.
Although the logic is correct, we used the wrong name here. See L669 in the constructor for the call.
clip_length_in_frames=self.num_frames, | ||
frames_between_clips=self.step, | ||
frame_rate=self.frame_rate, |
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.
I found this pretty hard to read without these being keyword args.
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 Philip, LGTM when green
Hey @pmeier! You merged this PR, but no labels were added. The list of valid labels is available at https://github.com/pytorch/vision/blob/main/.github/process_commit.py |
Summary: (Note: this ignores all push blocking failures!) Reviewed By: matteobettini Differential Revision: D48900404 fbshipit-source-id: 015ca49dd030e3734ad7a9d088ea454ffc2bf018
While looking into #7875, I realized we don't test for the correct output format of our video datasets. This PR adds a small test for that as well some QoL improvements I found along the way.