diff options
| author | Hiroshiba Kazuyuki <kazuyuki_hiroshiba@dwango.co.jp> | 2018-01-15 04:17:10 +0900 |
|---|---|---|
| committer | Hiroshiba Kazuyuki <kazuyuki_hiroshiba@dwango.co.jp> | 2018-01-15 04:41:33 +0900 |
| commit | 7bfc3321e356f24f49c790b578917e8db22bd30d (patch) | |
| tree | 65e8070b34d22e8b6211bb41e7dd448eb39dccd1 /scripts/extract_spectrogram_pair.py | |
| parent | 2be3f03adc5695f82c6ab86da780108f786ed014 (diff) | |
超解像学習を可能に
Diffstat (limited to 'scripts/extract_spectrogram_pair.py')
| -rw-r--r-- | scripts/extract_spectrogram_pair.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/scripts/extract_spectrogram_pair.py b/scripts/extract_spectrogram_pair.py index 02108fc..be21459 100644 --- a/scripts/extract_spectrogram_pair.py +++ b/scripts/extract_spectrogram_pair.py @@ -10,6 +10,7 @@ from pprint import pprint import numpy import pysptk import pyworld +from tqdm import tqdm from become_yukarin.dataset.dataset import AcousticFeatureProcess from become_yukarin.dataset.dataset import WaveFileLoadProcess @@ -68,7 +69,6 @@ def generate_file(path): 'low': low_spectrogram, 'high': high_spectrogram, }) - print('saved!', out) def main(): @@ -76,7 +76,7 @@ def main(): arguments.output_directory.mkdir(exist_ok=True) pool = multiprocessing.Pool() - pool.map(generate_file, paths) + list(tqdm(pool.imap(generate_file, paths), total=len(paths))) if __name__ == '__main__': |