o þϯiª ã@sªddlmZmZmZmZmZddlZddlmZddl m Z m Z  ej j j ddddfd ed ed eej j jd eeeeefgeeeffd ededede fdd„ZdS)é)ÚAnyÚCallableÚDictÚOptionalÚTypeN)Ú ClipSampleré)ÚLabeledVideoDatasetÚlabeled_video_datasetÚTÚpyavÚ data_pathÚ clip_samplerÚ video_samplerÚ transformÚvideo_path_prefixÚ decode_audioÚdecoderÚreturncCs tj d¡t|||||||ƒS)aÌ A helper function to create ``LabeledVideoDataset`` object for the Ucf101 dataset. Args: data_path (str): Path to the data. The path type defines how the data should be read: * For a file path, the file is read and each line is parsed into a video path and label. * For a directory, the directory structure defines the classes (i.e. each subdirectory is a class). clip_sampler (ClipSampler): Defines how clips should be sampled from each video. See the clip sampling documentation for more information. video_sampler (Type[torch.utils.data.Sampler]): Sampler for the internal video container. This defines the order videos are decoded and, if necessary, the distributed split. transform (Callable): This callable is evaluated on the clip output before the clip is returned. It can be used for user defined preprocessing and augmentations to the clips. See the ``LabeledVideoDataset`` class for clip output format. video_path_prefix (str): Path to root directory with the videos that are loaded in ``LabeledVideoDataset``. All the video paths before loading are prefixed with this path. decode_audio (bool): If True, also decode audio from video. decoder (str): Defines what type of decoder used to decode a video. zPYTORCHVIDEO.dataset.Ucf101)ÚtorchÚ_CÚ_log_api_usage_oncer )r rrrrrr©rúL/home/ubuntu/.local/lib/python3.10/site-packages/pytorchvideo/data/ucf101.pyÚUcf101s +ùr)ÚtypingrrrrrrÚpytorchvideo.data.clip_samplingrr r ÚutilsÚdataÚ RandomSamplerÚstrÚSamplerÚboolrrrrrÚs6  ùÿþ ý üûúùø