~VjiddlZddlmZddlmZmZddlZddlmZddl m Z ddl m Z ddl mZdZd Zd d d d ddddZdedededededeeeeeeeeff dZGdde ZdS)N)Path)TupleUnion)Tensor)Dataset)download_url_to_file) _extract_tartrain-clean-100LibriTTS@da0864e1bd26debed35da8a869dd5c04dfc27682921936de7cff9c8a254dbe1a@d413eda26f3a152ac7c9cf3658ef85504dfb1b625296e5fa83727f5186cca79c@234ea5b25859102a87024a4b9b86641f5b5aaaf1197335c95090cde04fe9a4f5@33a5342094f3bba7ccc2e0500b9e72d558f72eb99328ac8debe1d9080402f10d@c5608bf1ef74bb621935382b8399c5cdd51cd3ee47cec51f00f885a64c6c7f6b@ce7cff44dcac46009d18379f37ef36551123a1dc4e5c8e4eb73ae57260de4886@e35f7e34deeb2e2bdfe4403d88c8fdd5fbf64865cae41f027a185a6965f0a5df)z4http://www.openslr.org/resources/60/dev-clean.tar.gzz4http://www.openslr.org/resources/60/dev-other.tar.gzz5http://www.openslr.org/resources/60/test-clean.tar.gzz5http://www.openslr.org/resources/60/test-other.tar.gzz:http://www.openslr.org/resources/60/train-clean-100.tar.gzz:http://www.openslr.org/resources/60/train-clean-360.tar.gzz:http://www.openslr.org/resources/60/train-other-500.tar.gzfileidpath ext_audioext_original_txtext_normalized_txtreturnc|d\}}}}|}||z} tj|||| } ||z} tj|||| } ||z} tj|||| } t j| \} } t | 5}|} dddn #1swxYwYt | d5}|} dddn #1swxYwY| | | | t|t||fS)N_r) splitosrjoin torchaudioloadopenreadlineint)rrrrr speaker_id chapter_id segment_id utterance_idnormalized_text original_text file_audiowaveform sample_ratefts V/root/voice-cloning/.venv/lib/python3.11/site-packages/torchaudio/datasets/libritts.pyload_libritts_itemr/s8>||C7H7H4J J L"%77Ogll4ZQQO #33MGLLz:}MMM )JdJ JGGJ'OJ77Hk m  & &&&&&&&&&&&&&&& os # #(r++--(((((((((((((((  J J s$8CC C4DDDc eZdZdZdZdZdZeedfde e e fde de d e d d f d Z d ed eeee e eee ffdZd efdZd S)LIBRITTSa*LibriTTS* :cite:`Zen2019LibriTTSAC` 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 ``"dev-clean"``, ``"dev-other"``, ``"test-clean"``, ``"test-other"``, ``"train-clean-100"``, ``"train-clean-360"`` and ``"train-other-500"``. (default: ``"train-clean-100"``) folder_in_archive (str, optional): The top-level directory of the dataset. (default: ``"LibriTTS"``) download (bool, optional): Whether to download the dataset if it is not found at root path. (default: ``False``). z .original.txtz.normalized.txtz.wavFrooturlfolder_in_archivedownloadrNc|dvr'd}d}tj|||z}tj|}tj|}tj||}|dd}tj||}tj|||_|rtj|js[tj|s-t |d} t||| t|nK|]}t|jVdSN)strstem).0ps r. z$LIBRITTS.__init__..s*ddac!&kkddddddz*/*/*)rrrfspathbasenamer_pathisdirisfile _CHECKSUMSgetrr exists RuntimeErrorsortedrglob _ext_audio_walker) selfr2r3r4r5 ext_archivebase_urlrBarchivechecksums r.__init__zLIBRITTS.__init__Us    $K=H',,x{):;;Cy7##C((',,tX..>>#&&q)GLL):HEEW\\$(9::  7==,, &w~~g..M)~~c488H(g8LLLLW%%%7>>$*-- "[ [[[ dd4 3C3C3H3HSWSbIb3c3cddddd r@ncj|j|}t||j|j|j|jS)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: Original text str: Normalized text int: Speaker ID int: Chapter ID str: Utterance ID )rMr/rCrL_ext_original_txt_ext_normalized_txt)rNrTrs r. __getitem__zLIBRITTS.__getitem__s:0a!  J O  "  $    r@c*t|jSr:)lenrM)rNs r.__len__zLIBRITTS.__len__s4<   r@)__name__ __module__ __qualname____doc__rVrWrLURLFOLDER_IN_ARCHIVErr;rboolrSr#rrrXr[r@r.r1r1As  (+J !2 /e/eCI/e/e /e  /e  /e/e/e/eb S U63S#sC+O%P    B!!!!!!!r@r1)rpathlibrtypingrrrtorchrtorch.utils.datartorchaudio._internalrtorchaudio.datasets.utilsr r`rarFr;r#r/r1rcr@r.rjsc $$$$$$555555222222<~<~==CECECE & & && &  &  63S#sC /0 &&&&Rg!g!g!g!g!wg!g!g!g!g!r@