Yh@ d dl Z d dlZd dlmZ d dlmZmZmZmZ d dl m
Z
ddl m Z m
Z
ddlmZ ddlmZ G d d
e ZdS ) N)Path)AnyCallableOptionalUnion)Tensor ) find_classes make_dataset)
VideoClips)
VisionDatasetc ! f e Zd ZdZdZdddZdZdZ d$d ee e
f d
e de de de e de de
de e de ee ef de de de de de de ddf fd
Zedee ef fd Zd ee d e de de
dee f
d Zde fd!Zd"e deeee f fd#Z xZS )%HMDB51a
`HMDB51 `_
dataset.
HMDB51 is an action recognition video dataset.
This dataset consider every video as a collection of video clips of fixed size, specified
by ``frames_per_clip``, where the step in frames between each clip is given by
``step_between_clips``.
To give an example, for 2 videos with 10 and 15 frames respectively, if ``frames_per_clip=5``
and ``step_between_clips=5``, the dataset size will be (2 + 3) = 5, where the first two
elements will come from video 1, and the next three elements from video 2.
Note that we drop clips which do not have exactly ``frames_per_clip`` elements, so not all
frames in a video might be present.
Internally, it uses a VideoClips object to handle clip creation.
Args:
root (str or ``pathlib.Path``): Root directory of the HMDB51 Dataset.
annotation_path (str): Path to the folder containing the split files.
frames_per_clip (int): Number of frames in a clip.
step_between_clips (int): Number of frames between each clip.
fold (int, optional): Which fold to use. Should be between 1 and 3.
train (bool, optional): If ``True``, creates a dataset from the train split,
otherwise from the ``test`` split.
transform (callable, optional): A function/transform that takes in a TxHxWxC video
and returns a transformed version.
output_format (str, optional): The format of the output video tensors (before transforms).
Can be either "THWC" (default) or "TCHW".
Returns:
tuple: A 3-tuple with the following entries:
- video (Tensor[T, H, W, C] or Tensor[T, C, H, W]): The `T` video frames
- audio(Tensor[K, L]): the audio frames, where `K` is the number of channels
and `L` is the number of points
- label (int): class of the video clip
zJhttps://serre-lab.clps.brown.edu/wp-content/uploads/2013/10/hmdb51_org.rarzQhttps://serre-lab.clps.brown.edu/wp-content/uploads/2013/10/test_train_splits.rar 15e67781e70dcfbdce2d7dbb9b3344b5)urlmd5r NTr THWCrootannotation_pathframes_per_clipstep_between_clips
frame_ratefoldtrain transform_precomputed_metadata num_workers _video_width
_video_height_video_min_dimension_audio_samples
output_formatreturnc
t | |dvrt d| d}t | j \ | _ }t
| j || | _ d | j D }t ||||| |
| | |
|| }|| _ || _
|| _ | |||| | _
| | j
| _ || _ d S )N)r r z$fold should be between 1 and 3, got )avic g | ]\ }}|S r) ).0path_s m/var/www/tools.fuzzalab.pt/emblema-extractor/venv/lib/python3.11/site-packages/torchvision/datasets/hmdb51.py
z#HMDB51.__init__..[ s ::: qt::: )r r r r! r" r# )super__init__
ValueErrorr
r classesr samplesr full_video_clipsr r _select_foldindicessubset video_clipsr )selfr r r r r r r r r r r r r! r" r#
extensions class_to_idx video_pathsr9 __class__s r- r1 zHMDB51.__init__= s $ y JD J JKK K
%1$)%<%<" l# I
;:T\::: !#%'!5)'
!,
((otUSS &--dl;;"r/ c | j j S N)r5 metadatar: s r- rA zHMDB51.metadatas s $--r/
video_listannotations_dirc \ |r| j n| j }d| d}t j || }t j | }t
} |D ]}
t |
5 } | } d d d n # 1 swxY w Y | D ]C}
|
\ }}t | }||k r| | Dg }t | D ];\ }}t j
| | v r| | <|S )Nz *test_splitz.txt) TRAIN_TAGTEST_TAGosr+ joinglobsetopen readlinessplitintadd enumeratebasenameappend)r: rC rD r r
target_tagsplit_pattern_namesplit_pattern_pathannotation_pathsselected_filesfilepathfidlineslinevideo_filename
tag_stringtagr7 video_index
video_paths r- r6 zHMDB51._select_foldw sw ',?T^^$-
54 5 5 5 W\\/;M N N 9%788 ( 7 7Hh
(3
(
(
(
(
(
(
(
(
(
(
(
(
(
(
(
7
7-1ZZ\\*
*oo*$$"&&~666
7 '0'<'< , , #Kw
++~=={+++s ,B
B B c 4 | j S r@ )r9 num_clipsrB s r- __len__zHMDB51.__len__ s ))+++r/ idxc | j | \ }}}}| j | }| j | \ }}| j | | }|||fS r@ )r9 get_clipr7 r4 r )r: re videoaudior, video_idx sample_index class_indexs r- __getitem__zHMDB51.__getitem__ sc %)%5%>%>s%C%C"ua|I. l3; > %NN5))Ee[((r/ ) r Nr TNNr r r r r r )__name__
__module__ __qualname____doc__data_urlsplitsrF rG r strr rO r boolr dictr r1 propertyrA listr6 rd tupler rm
__classcell__)r> s @r- r r
s % %N \Hb1 F IH #$$((,:> $% #!4# 4#CI 4# 4# 4#
4# SM
4# 4# 4# H%4# (S#X 74# 4# 4# 4# " 4# 4# !4#"
#4# 4# 4# 4# 4# 4#l .$sCx. . . . X.tCy 3 c Z^ cghkcl ,, , , , ,)s )uVVS-@'A ) ) ) ) ) ) ) )r/ r )rJ rH pathlibr typingr r r r torchr folderr
r video_utilsr visionr
r r) r/ r- r s 1 1 1 1 1 1 1 1 1 1 1 1 . . . . . . . . # # # # # # ! ! ! ! ! !K) K) K) K) K)] K) K) K) K) K)r/