prepro added

Author: Kyubyong

data_load.py

@@ -100,7 +100,17 @@ def get_batch():
 
         # Parse
         text = tf.decode_raw(text, tf.int32)  # (None,)
-        fname, mel, mag = tf.py_func(load_spectrograms, [fpath], [tf.string, tf.float32, tf.float32])  # (None, n_mels)
+
+        if hp.prepro:
+            def _load_spectrograms(fpath):
+                fname = os.path.basename(fpath)
+                mel = "mels/{}".format(fname.replace("wav", "npy"))
+                mag = "mags/{}".format(fname.replace("wav", "npy"))
+                return fname, np.load(mel), np.load(mag)
+
+            fname, mel, mag = tf.py_func(_load_spectrograms, [fpath], [tf.string, tf.float32, tf.float32])
+        else:
+            fname, mel, mag = tf.py_func(load_spectrograms, [fpath], [tf.string, tf.float32, tf.float32])  # (None, n_mels)
 
         # Add shape information
         fname.set_shape(())

prepo.py

@@ -0,0 +1,25 @@
+# -*- coding: utf-8 -*-
+#/usr/bin/python2
+'''
+By kyubyong park. kbpark.linguist@gmail.com.
+https://www.github.com/kyubyong/dc_tts
+'''
+
+from __future__ import print_function
+
+from utils import load_spectrograms
+import os
+from data_load import load_data
+import numpy as np
+import tqdm
+
+# Load data
+fpaths, _, _ = load_data() # list
+
+for fpath in tqdm.tqdm(fpaths):
+    fname, mel, mag = load_spectrograms(fpath)
+    if not os.path.exists("mels"): os.mkdir("mels")
+    if not os.path.exists("mags"): os.mkdir("mags")
+
+    np.save("mels/{}".format(fname.replace("wav", "npy")), mel)
+    np.save("mags/{}".format(fname.replace("wav", "npy")), mag)