def _save_orig_plot_if_not_exists(dest_dir: str, mel):
path = os.path.join(dest_dir, "original.png")
if not os.path.isfile(path):
plot_melspec(mel, title="Original")
plt.savefig(path, bbox_inches='tight')
plt.close()
return path
def save_plot(dest_dir: str, data_len: int, wav_entry: WavData,
ds_entry: DsData, mel_tensor: Tensor) -> str:
chunk_dir = os.path.join(
dest_dir,
get_chunk_name(wav_entry.entry_id,
chunksize=CHUNK_SIZE,
maximum=data_len - 1))
os.makedirs(chunk_dir, exist_ok=True)
plot_melspec(mel_tensor, title=f"{repr(wav_entry)}: {ds_entry.text}")
path = os.path.join(chunk_dir, f"{repr(wav_entry)}.png")
plt.savefig(path, bbox_inches='tight')
plt.close()
return path
def save_infer_alignments_sentence_plot(infer_dir: str, infer_res: InferenceResult):
plot_melspec(infer_res.alignments, title="{}: Alignments {}".format(
get_parent_dirname(infer_dir), infer_res.sentence.sent_id))
path = os.path.join(infer_dir, f"{infer_res.sentence.sent_id}_alignments.png")
plt.savefig(path, bbox_inches='tight')
plt.close()
def save_infer_pre_postnet_sentence_plot(infer_dir: str, infer_res: InferenceResult):
plot_melspec(infer_res.mel_outputs_postnet,
title="{}: Pre-Postnet {}".format(get_parent_dirname(infer_dir), infer_res.sentence.sent_id))
path = os.path.join(infer_dir, f"{infer_res.sentence.sent_id}_pre_post.png")
plt.savefig(path, bbox_inches='tight')
plt.close()
def save_val_alignments_sentence_plot(val_dir: str, mel):
parent_dir = get_parent_dirname(val_dir)
plot_melspec(mel, title=f"{parent_dir}: Alignments")
path = os.path.join(val_dir, f"{parent_dir}_alignments.png")
plt.savefig(path, bbox_inches='tight')
plt.close()
def save_val_pre_postnet_plot(val_dir: str, mel):
parent_dir = get_parent_dirname(val_dir)
plot_melspec(mel, title=f"{parent_dir}: Pre-Postnet")
path = os.path.join(val_dir, f"{parent_dir}_pre_post.png")
plt.savefig(path, bbox_inches='tight')
plt.close()
def _save_trimmed_plot_temp(mel):
path = tempfile.mktemp(suffix=".png")
plot_melspec(mel, title="Trimmed")
plt.savefig(path, bbox_inches='tight')
plt.close()
return path
def save_infer_orig_plot(infer_dir: str, mel: np.ndarray):
plot_melspec(mel, title="Original")
path = os.path.join(infer_dir, f"{get_parent_dirname(infer_dir)}_orig.png")
plt.savefig(path, bbox_inches='tight')
plt.close()
return path
def save_val_orig_plot(val_dir: str, mel):
parent_dir = get_parent_dirname(val_dir)
plot_melspec(mel, title=parent_dir)
path = os.path.join(val_dir, f"{parent_dir}_orig.png")
plt.savefig(path, bbox_inches='tight')
plt.close()
