Symmetric mels helps quick alignment

Signed-off-by: begeekmyfriend <[email protected]>
begeekmyfriend · Dec 1, 2018 · bd2d80e · bd2d80e
1 parent 3765d6c
commit bd2d80e
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 5 deletions.
diff --git a/hparams.py b/hparams.py
@@ -8,7 +8,7 @@
   cleaners='basic_cleaners',
 
   # Audio:
-  num_mels=80,
+  num_mels=160,
   num_freq=1025,
   sample_rate=16000,
   frame_length_ms=50,
@@ -17,6 +17,7 @@
   min_level_db=-100,
   ref_level_db=20,
   max_frame_num=1000,
+  max_abs_value = 4,
 
   # Model:
   outputs_per_step=5,
@@ -38,7 +39,7 @@
   # Eval:
   max_iters=300,
   griffin_lim_iters=60,
-  power=1.5,              # Power to raise magnitudes to prior to Griffin-Lim
+  power=1.2,              # Power to raise magnitudes to prior to Griffin-Lim
 )
 
 

diff --git a/util/audio.py b/util/audio.py
@@ -144,10 +144,13 @@ def _db_to_amp_tensorflow(x):
   return tf.pow(tf.ones(tf.shape(x)) * 10.0, x * 0.05)
 
 def _normalize(S):
-  return (S - hparams.min_level_db) / -hparams.min_level_db
+  # symmetric mels
+  return 2 * hparams.max_abs_value * ((S - hparams.min_level_db) / -hparams.min_level_db) - hparams.max_abs_value
 
 def _denormalize(S):
-  return (S * -hparams.min_level_db) + hparams.min_level_db
+  # symmetric mels
+  return ((S + hparams.max_abs_value) * -hparams.min_level_db) / (2 * hparams.max_abs_value) + hparams.min_level_db
 
 def _denormalize_tensorflow(S):
-  return (S * -hparams.min_level_db) + hparams.min_level_db
+  # symmetric mels
+  return ((S + hparams.max_abs_value) * -hparams.min_level_db) / (2 * hparams.max_abs_value) + hparams.min_level_db