From 7bfc3321e356f24f49c790b578917e8db22bd30d Mon Sep 17 00:00:00 2001
From: Hiroshiba Kazuyuki <kazuyuki_hiroshiba@dwango.co.jp>
Date: Mon, 15 Jan 2018 04:17:10 +0900
Subject: 超解像学習を可能に
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 scripts/extract_spectrogram_pair.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

(limited to 'scripts/extract_spectrogram_pair.py')

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__':
-- 
cgit v1.2.3-70-g09d2