File size: 1,452 Bytes
b78ae92
 
 
 
 
 
 
 
56f8385
 
979ff6e
b78ae92
 
 
 
 
 
 
713281e
56f8385
 
 
 
 
 
 
 
 
 
713281e
b1ba841
56f8385
 
 
b1ba841
56f8385
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
import os

import torchaudio

from api import TextToSpeech
from utils.audio import load_audio

if __name__ == '__main__':
    fname = 'Y:\\clips\\books2\\subset512-oco.tsv'
    stop_after = 128
    outpath_base = 'D:\\tmp\\tortoise-tts-eval\\audiobooks'
    outpath_real = 'D:\\tmp\\tortoise-tts-eval\\real'

    os.makedirs(outpath_real, exist_ok=True)
    with open(fname, 'r', encoding='utf-8') as f:
        lines = [l.strip().split('\t') for l in f.readlines()]

    tts = TextToSpeech()
    for k in range(3):
        outpath = f'{outpath_base}_{k}'
        os.makedirs(outpath, exist_ok=True)
        recorder = open(os.path.join(outpath, 'transcript.tsv'), 'w', encoding='utf-8')
        for e, line in enumerate(lines):
            if e >= stop_after:
                break
            transcript = line[0]
            path = os.path.join(os.path.dirname(fname), line[1])
            cond_audio = load_audio(path, 22050)
            torchaudio.save(os.path.join(outpath_real, os.path.basename(line[1])), cond_audio, 22050)
            sample = tts.tts_with_preset(transcript, [cond_audio, cond_audio], preset='standard')

            down = torchaudio.functional.resample(sample, 24000, 22050)
            fout_path = os.path.join(outpath, os.path.basename(line[1]))
            torchaudio.save(fout_path, down.squeeze(0), 22050)

            recorder.write(f'{transcript}\t{fout_path}\n')
            recorder.flush()
        recorder.close()