Skip to content

Commit 5ef75d7

Browse files
Update UCF101 docstring (#4275)
* Update UCF101 docstring * addressing @NicolasHugs comments * Fixed flake8 error Co-authored-by: Prabhat Roy <[email protected]>
1 parent a3d9c41 commit 5ef75d7

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

torchvision/datasets/ucf101.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,9 @@ class UCF101(VisionDataset):
1414
UCF101 is an action recognition video dataset.
1515
This dataset consider every video as a collection of video clips of fixed size, specified
1616
by ``frames_per_clip``, where the step in frames between each clip is given by
17-
``step_between_clips``.
17+
``step_between_clips``. The dataset itself can be downloaded from the dataset website;
18+
annotations that ``annotation_path`` should be pointing to can be downloaded from `here
19+
<https://www.crcv.ucf.edu/data/UCF101/UCF101TrainTestSplits-RecognitionTask.zip>`.
1820
1921
To give an example, for 2 videos with 10 and 15 frames respectively, if ``frames_per_clip=5``
2022
and ``step_between_clips=5``, the dataset size will be (2 + 3) = 5, where the first two
@@ -26,7 +28,8 @@ class UCF101(VisionDataset):
2628
2729
Args:
2830
root (string): Root directory of the UCF101 Dataset.
29-
annotation_path (str): path to the folder containing the split files
31+
annotation_path (str): path to the folder containing the split files;
32+
see docstring above for download instructions of these files
3033
frames_per_clip (int): number of frames in a clip.
3134
step_between_clips (int, optional): number of frames between each clip.
3235
fold (int, optional): which fold to use. Should be between 1 and 3.

0 commit comments

Comments
 (0)