facebookresearch / pytorchvideo

A deep learning library for video understanding research.

Home Page:https://pytorchvideo.org/

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

add support for creating LabeledVideoDataset from video frames

fcakyon opened this issue · comments

Currently, Video object can be constructed from a video folder that contains video frames:

def from_frame_paths(

However, LabeledVideoDataset cannot be created from video frames since LabeledVideoPaths class does not support that:

labeled_video_paths = LabeledVideoPaths.from_path(data_path)

To solve this issue, from_path method of the LabeledVideoPaths

def from_path(cls, data_path: str) -> LabeledVideoPaths:
can be extended so that it can handle video frames.

Thanks for working this! I will review the PR soon.