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