🚀 Enhance tacotron2 dataset to adapt with any reduction factor, apply black.

Author: dathudeptrai

examples/fastspeech/decode_fastspeech.py

@@ -33,32 +33,60 @@ def main():
     """Run fastspeech decoding from folder."""
     parser = argparse.ArgumentParser(
         description="Decode soft-mel features from charactor with trained FastSpeech "
-                    "(See detail in examples/fastspeech/decode_fastspeech.py).")
-    parser.add_argument("--rootdir", default=None, type=str, required=True,
-                        help="directory including ids/durations files.")
-    parser.add_argument("--outdir", type=str, required=True,
-                        help="directory to save generated speech.")
-    parser.add_argument("--checkpoint", type=str, required=True,
-                        help="checkpoint file to be loaded.")
-    parser.add_argument("--config", default=None, type=str, required=True,
-                        help="yaml format configuration file. if not explicitly provided, "
-                             "it will be searched in the checkpoint directory. (default=None)")
-    parser.add_argument("--batch-size", default=8, type=int, required=False,
-                        help="Batch size for inference.")
-    parser.add_argument("--verbose", type=int, default=1,
-                        help="logging level. higher is more logging. (default=1)")
+        "(See detail in examples/fastspeech/decode_fastspeech.py)."
+    )
+    parser.add_argument(
+        "--rootdir",
+        default=None,
+        type=str,
+        required=True,
+        help="directory including ids/durations files.",
+    )
+    parser.add_argument(
+        "--outdir", type=str, required=True, help="directory to save generated speech."
+    )
+    parser.add_argument(
+        "--checkpoint", type=str, required=True, help="checkpoint file to be loaded."
+    )
+    parser.add_argument(
+        "--config",
+        default=None,
+        type=str,
+        required=True,
+        help="yaml format configuration file. if not explicitly provided, "
+        "it will be searched in the checkpoint directory. (default=None)",
+    )
+    parser.add_argument(
+        "--batch-size",
+        default=8,
+        type=int,
+        required=False,
+        help="Batch size for inference.",
+    )
+    parser.add_argument(
+        "--verbose",
+        type=int,
+        default=1,
+        help="logging level. higher is more logging. (default=1)",
+    )
     args = parser.parse_args()
 
     # set logger
     if args.verbose > 1:
         logging.basicConfig(
-            level=logging.DEBUG, format="%(asctime)s (%(module)s:%(lineno)d) %(levelname)s: %(message)s")
+            level=logging.DEBUG,
+            format="%(asctime)s (%(module)s:%(lineno)d) %(levelname)s: %(message)s",
+        )
     elif args.verbose > 0:
         logging.basicConfig(
-            level=logging.INFO, format="%(asctime)s (%(module)s:%(lineno)d) %(levelname)s: %(message)s")
+            level=logging.INFO,
+            format="%(asctime)s (%(module)s:%(lineno)d) %(levelname)s: %(message)s",
+        )
     else:
         logging.basicConfig(
-            level=logging.WARN, format="%(asctime)s (%(module)s:%(lineno)d) %(levelname)s: %(message)s")
+            level=logging.WARN,
+            format="%(asctime)s (%(module)s:%(lineno)d) %(levelname)s: %(message)s",
+        )
         logging.warning("Skip DEBUG/INFO messages")
 
     # check directory existence
@@ -81,12 +109,14 @@ def main():
         root_dir=args.rootdir,
         charactor_query=char_query,
         charactor_load_fn=char_load_fn,
-        return_utt_id=True
+        return_utt_id=True,
     )
     dataset = dataset.create(batch_size=args.batch_size)
 
     # define model and load checkpoint
-    fastspeech = TFFastSpeech(config=FastSpeechConfig(**config["fastspeech_params"]), name='fastspeech')
+    fastspeech = TFFastSpeech(
+        config=FastSpeechConfig(**config["fastspeech_params"]), name="fastspeech"
+    )
     fastspeech._build()
     fastspeech.load_weights(args.checkpoint)
 
@@ -107,15 +137,22 @@ def main():
         masked_mel_afters = masked_mel_after.numpy()
 
         for (utt_id, mel_before, mel_after, durations) in zip(
-                utt_ids, masked_mel_befores, masked_mel_afters, duration_outputs):
+            utt_ids, masked_mel_befores, masked_mel_afters, duration_outputs
+        ):
             # real len of mel predicted
             real_length = durations.numpy().sum()
             utt_id = utt_id.numpy().decode("utf-8")
             # save to folder.
-            np.save(os.path.join(args.outdir, f"{utt_id}-fs-before-feats.npy"),
-                    mel_before[:real_length, :].astype(np.float32), allow_pickle=False)
-            np.save(os.path.join(args.outdir, f"{utt_id}-fs-after-feats.npy"),
-                    mel_after[:real_length, :].astype(np.float32), allow_pickle=False)
+            np.save(
+                os.path.join(args.outdir, f"{utt_id}-fs-before-feats.npy"),
+                mel_before[:real_length, :].astype(np.float32),
+                allow_pickle=False,
+            )
+            np.save(
+                os.path.join(args.outdir, f"{utt_id}-fs-after-feats.npy"),
+                mel_after[:real_length, :].astype(np.float32),
+                allow_pickle=False,
+            )
 
 
 if __name__ == "__main__":