def _save_orig_plot_if_not_exists(dest_dir: Path, mel) -> Path: path = dest_dir / "original.png" if not path.is_file(): plot_melspec(mel, title="Original") plt.savefig(path, bbox_inches='tight') plt.close() return path
def save_plot(dest_dir: Path, data_len: int, wav_entry: WavData, ds_entry: DsData, mel_tensor: Tensor) -> Path: chunk_dir = dest_dir / get_chunk_name(wav_entry.entry_id, chunksize=DEFAULT_PRE_CHUNK_SIZE, maximum=data_len - 1) chunk_dir.mkdir(parents=True, exist_ok=True) plot_melspec(mel_tensor, title=f"{repr(wav_entry)}: {ds_entry.text}") absolute_path = chunk_dir / f"{repr(wav_entry)}.png" plt.savefig(absolute_path, bbox_inches='tight') plt.close() return absolute_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()
def save_infer_plot(infer_dir: str, mel: np.ndarray): plot_melspec(mel, title=get_parent_dirname(infer_dir)) path = os.path.join(infer_dir, f"{get_parent_dirname(infer_dir)}.png") plt.savefig(path, bbox_inches='tight') plt.close() return path
def _save_trimmed_plot_temp(mel: np.ndarray) -> Path: path = Path(tempfile.mktemp(suffix=".png")) plot_melspec(mel, title="Trimmed") plt.savefig(path, bbox_inches='tight') plt.close() return path