~VjigddlZddlZddlmZddlmZmZmZddlZddl m Z ddl m Z ddl mZmZdZdZd Zgd ZGd d e Zd ededefdZdS)N)Path)OptionalTupleUnion)Dataset)download_url_to_file) _extract_tar_load_waveformz6https://speech.fit.vutbr.cz/files/quesst14Database.tgzi@@4f869e06bc066bbe9c5dde31dbd3909a0870d70291110ebbb38878dcbc2fc5e4)albanianbasqueczech nnenglishromanianslovakc eZdZdZ ddeeefdedeededd f d Z d e de ee effd Z d e de e je effd Zde fdZd S)QUESST14a*QUESST14* :cite:`Mir2015QUESST2014EQ` dataset. Args: root (str or Path): Root directory where the dataset's top level directory is found subset (str): Subset of the dataset to use. Options: [``"docs"``, ``"dev"``, ``"eval"``]. language (str or None, optional): Language to get dataset for. Options: [``None``, ``albanian``, ``basque``, ``czech``, ``nnenglish``, ``romanian``, ``slovak``]. If ``None``, dataset consists of all languages. (default: ``"nnenglish"``) download (bool, optional): Whether to download the dataset if it is not found at root path. (default: ``False``) rFrootsubsetlanguagedownloadreturnNc|dvrtd|-|tvr$tdtttj|}tjt}tj||}| ddd}tj|||_ tj |j s\tj |s-|stdtt|tt!|||d krt#|j |d |_dS|d krt#|j |d |_dS|d krt#|j |d|_dSdS)N)docsdevevalz/`subset` must be one of ['docs', 'dev', 'eval']z"`language` must be None or one of .rz9Dataset not found. Please use `download=True` to download) hash_prefixrzlanguage_key_utterances.lstrzlanguage_key_dev.lstrzlanguage_key_eval.lst) ValueError _LANGUAGESstrosfspathpathbasenameURLjoinrsplit_pathisdirisfile RuntimeErrorr _CHECKSUMr filter_audio_pathsdata)selfrrrrr&archives V/root/voice-cloning/.venv/lib/python3.11/site-packages/torchaudio/datasets/quesst14.py__init__zQUESST14.__init__&s 0 0 0NOO O  HJ$>$>S#j//SSTT Ty7##C((',,tX..??3**1-W\\$11 w}}TZ(( (7>>'** Jd&'bccc$S'yIIII $ ' ' ' V  *4:xA^__DIII u__*4:xAWXXDIII v  *4:xAXYYDIII nc|j|}tj||j}|t |djfS)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 audio int: Sample rate str: File name )r0r#r%relpathr* SAMPLE_RATE with_suffixname)r1r6 audio_pathr9s r3 get_metadatazQUESST14.get_metadataJsD"Yq\ '//*dj99 Z%;%;B%?%?%DDDr5c||}t|j|d|d}|f|ddzS)a:Load 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: File name rN)r>r r*)r1r6metadatawaveforms r3 __getitem__zQUESST14.__getitem___sG $$Q''!$*hqk8A;GG{Xabb\))r5c*t|jS)N)lenr0)r1s r3__len__zQUESST14.__len__ss49~~r5)rF)__name__ __module__ __qualname____doc__rr"rrboolr4intrr>torchTensorrCrFr5r3rrs   #. "Z"ZCI"Z"Z3- "Z  "Z  "Z"Z"Z"ZHEcEeCcM&:EEEE**S*U5<c+A%B****(r5rr%rlst_namecNg}t|}t|dz |z 5}|D]b}|\}}|||kr4t jdd|}|||z c dddn #1swxYwY|S)z+Extract audio paths for the given language.scoringNz^.*?\/r8)ropenstripsplitresubappend)r%rrP audio_pathsfliner=langs r3r/r/ws K ::D dY) * *2a 2 2D#zz||1133 J#(8(8 2z::J   tj0 1 1 1 1  2222222222222222 sA&BB!B)r#rVpathlibrtypingrrrrMtorch.utils.datartorchaudio._internalrtorchaudio.datasets.utilsr r r'r:r.r!rr"r/rOr5r3rbs )))))))))) $$$$$$555555BBBBBBBB? N  [[[[[w[[[| r5