Add resample script

TTS/bin/resample.py

@@ -0,0 +1,69 @@
+import argparse
+import glob
+import shutil
+import librosa
+from argparse import RawTextHelpFormatter
+from multiprocessing import Pool
+from tqdm import tqdm
+
+def resample_file(filename):
+    global args
+    y, sr = librosa.load(filename, sr=args.output_sr)
+    librosa.output.write_wav(filename, y, sr)
+
+if __name__ == '__main__':
+
+    parser = argparse.ArgumentParser(
+        description='''Resample a folder recusively with librosa
+Can be used in place or create a copy of the folder as an output.\n\n'''
+
+'''
+Example run:
+    python TTS/bin/resample.py
+        --input_dir /root/LJSpeech-1.1/
+        --output_sr 22050
+        --output_dir /root/resampled_LJSpeech-1.1/
+        --n_jobs 24
+''',
+        formatter_class=RawTextHelpFormatter)
+
+    parser.add_argument('--input_dir',
+                        type=str,
+                        default=None,
+                        required=True,
+                        help='Path of the folder containing the audio files to resample')
+    
+    parser.add_argument('--output_sr',
+                        type=int,
+                        default=22050,
+                        required=False,
+                        help='Samlple rate to which the audio files should be resampled')
+
+    parser.add_argument('--output_dir',
+                        type=str,
+                        default=None,
+                        required=False,
+                        help='Path of the destination folder. If not defined, the operation is done in place')
+    
+    parser.add_argument('--n_jobs',
+                        type=int,
+                        default=None,
+                        help='Number of threads to use, by default it uses all cores')
+
+    args = parser.parse_args()
+
+    if args.output_dir:
+        print('Recursively copying the input folder...')
+        shutil.copytree(args.input_dir, args.output_dir)
+        args.input_dir = args.output_dir
+
+    print('Resampling the audio files...')
+    audio_files = glob.glob(args.input_dir+'**/*.wav', recursive=True)
+    print(f'Found {len(audio_files)} files...')
+    with Pool(processes=args.n_jobs) as p:
+        with tqdm(total=len(audio_files)) as pbar:
+            for i, _ in enumerate(p.imap_unordered(resample_file, audio_files)):
+                pbar.update()
+
+    print('Done !')
+