~Vji9 ddlZddlmZddlmZmZmZddlmZddl m Z ddl m Z ddl mZmZdZd Zd Zd Zd Zd ddZdZdededeeeeeeffdZGdde ZdS)N)Path)OptionalTupleUnion)Tensor)Dataset)download_url_to_file) _extract_tar_load_waveformSpeechCommandsspeech_commands_v0.02_nohash__background_noise_i>@743935421bb51cccdb6bdd152e04c5c70274e935c82119ad7faeec31780d811d@af14739ee7dc311471de98f5f9d2c9191b18aedfe957f4a6ff791c709868ff58)z@http://download.tensorflow.org/data/speech_commands_v0.01.tar.gzz@http://download.tensorflow.org/data/speech_commands_v0.02.tar.gzcg}|D]Z}tj|}t|5}|fd|Dz }dddn #1swxYwY[|S)Nc g|]Q}tjtj|RS)ospathnormpathjoinstrip).0lineroots \/root/voice-cloning/.venv/lib/python3.11/site-packages/torchaudio/datasets/speechcommands.py z_load_list..sA^^^drw'' T4::<<(H(HII^^^)rrropen)r filenamesoutputfilenamefilepathfileobjs` r _load_listr&s F__7<<h// (^^ _w ^^^^V]^^^ ^F _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ MsAA A r$rreturnctj||}tj|\}}tj|\}}tj|\}}tj|\}}|t \}} t | } |t||| fSN)rrrelpathsplitsplitext HASH_DIVIDERint SAMPLE_RATE) r$rr*reldirr#_labelspeaker speaker_idutterance_numbers r_get_speechcommands_metadatar6sgooh--Gw}}W--FHw}}V$$HAu!!(++JGQ!!'**JGQ#*==#>#> J +,, K 4D DDrc eZdZdZeeddfdeeefdedede de ed df d Z d e d e ee eee ffd Zd e d e ee eee ffd Zd e fdZdS)SPEECHCOMMANDSa,*Speech Commands* :cite:`speechcommandsv2` dataset. Args: root (str or Path): Path to the directory where the dataset is found or downloaded. url (str, optional): The URL to download the dataset from, or the type of the dataset to dowload. Allowed type values are ``"speech_commands_v0.01"`` and ``"speech_commands_v0.02"`` (default: ``"speech_commands_v0.02"``) folder_in_archive (str, optional): The top-level directory of the dataset. (default: ``"SpeechCommands"``) download (bool, optional): Whether to download the dataset if it is not found at root path. (default: ``False``). subset (str or None, optional): Select a subset of the dataset [None, "training", "validation", "testing"]. None means the whole dataset. "validation" and "testing" are defined in "validation_list.txt" and "testing_list.txt", respectively, and "training" is the rest. Details for the files "validation_list.txt" and "testing_list.txt" are explained in the README of the dataset and in the introduction of Section 7 of the original paper and its reference 12. The original paper can be found `here `_. (Default: ``None``) FNrurlfolder_in_archivedownloadsubsetr'c ||dvrtd|dvr'd}d}tj|||z}tj|}tj|||_tj|}tj||} |ddd}tj||}tj|||_|rtj |jsatj | s-t |d} t|| | t| |jnc34K|]}t|VdSr)strrps r z*SPEECHCOMMANDS.__init__..|(MMqCFFMMMMMMrz*/*.wavczg|]7}t|v t|vtj|v5|8Sr)r- EXCEPT_FOLDERrrr)rwexcludess rrz+SPEECHCOMMANDS.__init__..}sT1$$a)?)?BGDTDTUVDWDW_gDgDgDgDgDgrc34K|]}t|VdSr)rErGs rrIz*SPEECHCOMMANDS.__init__..rJrc6g|]}t|v t|v|Sr)r-rL)rrMs rrz+SPEECHCOMMANDS.__init__..s-^^^!1B1B}\]G]G]AG]G]G]r) ValueErrorrrrfspath_archivebasenamersplit_pathisdirisfile _CHECKSUMSgetr r exists RuntimeErrorr&_walkersetsortedrglob) selfrr9r:r;r<base_url ext_archiverTarchivechecksumwalkerrNs @r__init__zSPEECHCOMMANDS.__init__Hs  &0U"U"Urss s    >H#K',,x{):;;Cy T+<== 7##C((',,tX..??3**1-GLL):HEEW\\$(9::  7==,, 2w~~g..M)~~c488H(g8LLLLWdj1117>>$*-- "[ [[[ \ ! !%dj2GHHDLLL y %dj2DEEDLLL z ! !:dj2GI[\\]]HMMD,<,<,A,A),L,LMMMMMFDLLL MMD,<,<,A,A),L,LMMMMMF^^v^^^DLLLrncF|j|}t||jS)aGet metadata for the n-th sample from the dataset. Returns filepath instead of waveform, but otherwise returns the same fields as :py:func:`__getitem__`. Args: n (int): The index of the sample to be loaded Returns: Tuple of the following items; str: Path to the audio int: Sample rate str: Label str: Speaker ID int: Utterance number )r]r6rS)rarhfileids r get_metadatazSPEECHCOMMANDS.get_metadatas!*a+FDMBBBrc||}t|j|d|d}|f|ddzS)aLoad the n-th sample from the dataset. Args: n (int): The index of the sample to be loaded Returns: Tuple of the following items; Tensor: Waveform int: Sample rate str: Label str: Speaker ID int: Utterance number rN)rkr rS)rarhmetadatawaveforms r __getitem__zSPEECHCOMMANDS.__getitem__sG($$Q''!$-!hqkJJ{Xabb\))rc*t|jSr))lenr])ras r__len__zSPEECHCOMMANDS.__len__s4<   r)__name__ __module__ __qualname____doc__URLFOLDER_IN_ARCHIVErrFrboolrrgr.rrkrrprsrrrr8r82s0!2 $ <_<_CI<_<_ <_  <_  <_ <_<_<_<_|CcCeCc3,C&DCCCC0*S*U63S#+E%F****0!!!!!!!rr8)rpathlibrtypingrrrtorchrtorch.utils.datartorchaudio._internalr torchaudio.datasets.utilsr r ryrxr-rLr/rYr&rFr.r6r8rrrrsO ))))))))))$$$$$$555555BBBBBBBB$ $  IKIK E3EcEeCcSVX[D[>\EEEE(E!E!E!E!E!WE!E!E!E!E!r