o *i9 @sddlZddlmZddlmZmZmZddlmZddl m Z ddl m Z ddl mZmZdZd Zd Zd Zd Zd ddZddZdededeeeeeeffddZGddde ZdS)N)Path)OptionalTupleUnion)Tensor)Dataset)download_url_to_file) _extract_tar_load_waveformZSpeechCommandsspeech_commands_v0.02Z_nohash_Z_background_noise_i>Z@743935421bb51cccdb6bdd152e04c5c70274e935c82119ad7faeec31780d811dZ@af14739ee7dc311471de98f5f9d2c9191b18aedfe957f4a6ff791c709868ff58)z@http://download.tensorflow.org/data/speech_commands_v0.01.tar.gzz@http://download.tensorflow.org/data/speech_commands_v0.02.tar.gzc s^g}|D](}tj|}t|}|fdd|D7}Wdn1s'wYq|S)Nc s&g|]}tjtj|qS)ospathnormpathjoinstrip).0linerootr q/var/www/html/eduruby.in/lip-sync/lip-sync-env/lib/python3.10/site-packages/torchaudio/datasets/speechcommands.py s&z_load_list..)r rropen)r filenamesoutputfilenamefilepathfileobjr rr _load_lists rrrreturnc Csrtj||}tj|\}}tj|\}}tj|\}}tj|\}}|t\}} t| } |t||| fSN)r rrelpathsplitsplitext HASH_DIVIDERint SAMPLE_RATE) rrr!Zreldirr_labelspeakerZ speaker_idZutterance_numberr r r_get_speechcommands_metadatasr*c @seZdZdZeeddfdeeefdedede de ed df d d Z d e d e ee eee ffd dZd e d e ee eee ffddZd e fddZdS)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_archivedownloadsubsetrc s|dur |dvr td|dvrd}d}tj|||}t|}tj|||_tj|}tj||} |ddd}tj||}tj|||_|rqtj |jsptj | sjt |d} t || | d t| |jntj|jstd |jd |d krt|jd |_dS|dkrt|jd|_dS|dkrtt|jd dtddt|jdD} fdd| D|_dStddt|jdD} dd| D|_dS)N)training validationtestingzSWhen `subset` is not None, it must be one of ['training', 'validation', 'testing'].)zspeech_commands_v0.01r z$http://download.tensorflow.org/data/z.tar.gz.r)Z hash_prefixz The path zT doesn't exist. Please check the ``root`` path or set `download=True` to download itr1zvalidation_list.txtr2ztesting_list.txtr0cs|]}t|VqdSr strrpr r r |z*SPEECHCOMMANDS.__init__..z*/*.wavcs0g|]}t|vrt|vrtj|vr|qSr )r$ EXCEPT_FOLDERr rrrwexcludesr rr}s  z+SPEECHCOMMANDS.__init__..csr5r r6r8r r rr:r;cSs g|] }t|vrt|vr|qSr )r$r<r=r r rrs ) ValueErrorr rrfspath_archivebasenamersplit_pathisdirisfile _CHECKSUMSgetrr exists RuntimeErrorr_walkersetsortedrglob) selfrr,r-r.r/base_urlZ ext_archiverDarchiveZchecksumZwalkerr r?r__init__HsH         zSPEECHCOMMANDS.__init__ncCs|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 )rMr*rC)rQrUZfileidr r r get_metadatas  zSPEECHCOMMANDS.get_metadatacCs2||}t|j|d|d}|f|ddS)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)rVr rC)rQrUmetadataZwaveformr r r __getitem__s zSPEECHCOMMANDS.__getitem__cCs t|jSr )lenrM)rQr r r__len__s zSPEECHCOMMANDS.__len__)__name__ __module__ __qualname____doc__URLFOLDER_IN_ARCHIVErr7rboolrrTr%rrVrrYr[r r r rr+2s,  > r+)r pathlibrtypingrrrZtorchrZtorch.utils.datarZtorchaudio._internalrZtorchaudio.datasets.utilsr r rar`r$r<r&rIrr7r%r*r+r r r rs$    $