]i&SrSSKJr SSKrSSKrSSKrSSKrSSKJr SSK r SSK r SSK r SSK r SSK Jr SSKrSSKJr SSKJr SS KJr S rS rS \4S jrS \4Sjr"SS5rSSjrSrSrg)z.Loading wav based datasets, including MusdbHQ.) OrderedDictN)Path) distributed) functional)convert_audio_channels)distribmixturez.wavTc 2SnSnSnSnU[/-GHnXU3- n U[:XaU R5(dSn UH'n X U3- n [R"U 5upX- n M) U R 5R 5S:nU(a [R "5S:XdS5e[R"XWSS9 [R"[U 55nURnUcUnURnOJUU:wa[SU SUS US 35eURU:wa[S U SUS URS 35eU[:XdGMMU(dGMW[R"[U 55unnURS5nUR5R!5nUR#5R!5nGM WXgUS .$![a [U 5 ef=f![a [U 5 ef=f) Nrr soundfilezuse dset.backend=soundfilePCM_F)encodingzInvalid length for file z : expecting z but got .zInvalid sample rate for file )lengthmeanstd samplerate)MIXTUREexiststaloadabsmaxget_audio_backendsaveinfostr RuntimeErrorprint num_frames sample_rate ValueErrorritemr)tracksources normalizeext track_lengthtrack_sampleraterrsourcefileaudio sub_sourcesub_file sub_audiosr would_cliprrwav_s G/mnt/rpi/tmp/demucs-venv-sys/lib/python3.13/site-packages/demucs/wav.py_track_metadatar5s"L D CWI%#'' W T[[]]E%  \##77 " 1 "&*a/J++-<Z>ZZ< GGDg 6 773t9%D  !L#//  V #*4&1)N)F81>? ?  !1 1/v6-.i8H8H7ILM M W  T+Q((1+C88:??$D'').."CM&PdFV WW3  $K  $  d  s G%+"G?%G<?Hc 0n[U5n/nSSKJn U"S5n[R"USS9Hupn [U5nUR R S5(d U (dX:XaM?[URU55n URXR[XX#545 M [R"USS9HupU R5XK'M S S S 5 U$!,(df  U$=f) a5 Build the metadata for `Wavset`. Args: path (str or Path): path to dataset. sources (list[str]): list of sources to look for. normalize (bool): if True, loads full track and store normalization values based on the mixture file. ext (str): extension of audio files (default is .wav). r)ThreadPoolExecutorT) followlinksrx)ncolsN)rconcurrent.futuresr7oswalkname startswithr relative_toappendsubmitr5tqdmresult) pathr%r&r'metapendingsr7poolrootfoldersfilesr?pendings r4build_metadatarNNs D :DH5 A $$&GGDd$C D5:Dyy##C((Gt|t''-.D OOT;;y#^_ ` %D"YYxs;MD )DJ<  K   Ks C C22 Dc<\rSrSrSSSSS\4SjrSrSrS rS r g) WavsetkNTiDc [U5Ul[U5UlX@lU=(d UUlX`lX0lXlXpl Xl /Ul URR5HqupU SU S- n UbX:aSn O;[[R"XR- UR - 5S-5n URR!U 5 Ms g)a Waveset (or mp3 set for that matter). Can be used to train with arbitrary sources. Each track should be one folder inside of `path`. The folder should contain files named `{source}.{ext}`. Args: root (Path or str): root folder for the dataset. metadata (dict): output from `build_metadata`. sources (list[str]): list of source names. segment (None or float): segment length in seconds. If `None`, returns entire tracks. shift (None or float): stride in seconds bewteen samples. normalize (bool): normalizes input audio, **based on the metadata content**, i.e. the entire track is normalized, not individual extracts. samplerate (int): target sample rate. if the file sample rate is different, it will be resampled on the fly. channels (int): target nb of channels. if different, will be changed onthe fly. ext (str): extension for audio files (default is .wav). samplerate and channels are converted on the fly. rrNr)rrJrmetadatasegmentshiftr&r%channelsrr' num_examplesitemsintmathceilrB)selfrJrTr%rUrVr&rrWr'r?rGtrack_durationexampless r4__init__Wavset.__init__ls4J #H-  %g "  $----/JD!(^d<.@@N.":tyy.<<*G4::)UVYZZ[    $ $X . 0c,[UR5$N)sumrX)r]s r4__len__Wavset.__len__s4$$%%rbcBURU- UUR3- $rd)rJr')r]r?r*s r4get_fileWavset.get_files#yy4VHTXXJ"777rbc[URUR5GHup#X:aX-nMURUnSnSnURbM[ USUR -U-5n[ [ R"USUR-55n/nURHZnURX(5n [R"[U 5XeS9up[XR5n URU 5 M\ [ R""U5n [$R&"XSUR(5n UR*(a XS- US- n UR(aS[ URUR(-5n U SSU 24n [,R."U SXR0S- 45n U s $ g)Nrr) frame_offsetr rr.)ziprTrXrUrZrVr[r\r%rirrrrrWrBthstackjulius resample_fracrr&Fpadshape)r]indexr?r_rGr offsetwavsr*r+r2r3examplers r4 __getitem__Wavset.__getitem__s~!$--1B1BCND !==&DJF||'T,/$**z$,,#||4::"&//DMM!%1Iz7)d4<<6H*H:"&//DMM!%:38:I  rbcSSKn[[RS5S- S- nUR [ US55nUS$)Nrconfigszmus.yamlrvalidation_tracks)yamlrmusdb__path__ safe_loadr)r setup_pathsetups r4_get_musdb_validrsCennQ'(94zAJ NN4 C0 1E $ %%rbc  [R"[UR5R 55R 5SSn[ UR5SU-S-- n[ UR5S- nUR5(dd[RS:XaPURRSSS9 [X0R5n[R "U[#US 55 [R$S :a[&R("5 [R*"[#U55n[-5nUR.(aUnO+UR15VVs0sHupxXu;dM Xx_M nnnUR15VVs0sHupxXu;dM Xx_M n nnUR2(a0n OUR4UR6S .n [9X6URUR4UR6UR:UR<UR>S 9n [9X9[@/[CUR5-4UR:UR<UR>S .U D6n X4$s snnfs snnf)z0Extract the musdb dataset from the XP arguments.Nr8musdb_rrrTrrrrrr)"rrrrrrrrTrr rrrrNr%rrrrrrrr train_validrYrrUrVrPrrWr&rr) rrrrJrT valid_tracksmetadata_trainr?rGmetadata_validrrrs r4get_musdb_wav_datasetsrs ,,s4::--/ 0 : : G+CDM  g %D  " "w||q'8""D$"?!$ 5 (D45Ayym,-H#%L !7?~~7Gd7G4Kc*$*7Gd3;>>3C\3CZTtG[jdj3CN\ || LL4::>tT\\#||4::"&//DMM!%1ItgYdll9K-K:"&//DMM!%:38:I  e\s1 I9I9 I?+I?)r2) __doc__ collectionsrrr[rr=pathlibrrDrrqtorchror torchaudiortorch.nnrrsr,rr rrr5rNrPrrrrrbr4rsw 5#  $)  /3-X`-1c:MM` :& rb