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()