def save_v(infer_dir: str):
path1 = os.path.join(infer_dir,
f"{get_parent_dirname(infer_dir)}_orig.png")
path2 = os.path.join(infer_dir, f"{get_parent_dirname(infer_dir)}.png")
path3 = os.path.join(infer_dir,
f"{get_parent_dirname(infer_dir)}_diff.png")
path = os.path.join(infer_dir, f"{get_parent_dirname(infer_dir)}_v.png")
stack_images_vertically([path1, path2, path3], path)
def save_val_comparison(val_dir: str):
parent_dir = get_parent_dirname(val_dir)
path1 = os.path.join(val_dir, f"{parent_dir}_orig.png")
path2 = os.path.join(val_dir, f"{parent_dir}.png")
assert os.path.exists(path1)
assert os.path.exists(path2)
path = os.path.join(val_dir, f"{parent_dir}_comp.png")
stack_images_vertically([path1, path2], path)
def plot_batches_h_v(batches, plots_dir):
for i, v_batch in enumerate(tqdm(batches)):
h_files = []
for h_batch in v_batch:
h_path = mktemp(suffix=".png")
stack_images_horizontally(h_batch, h_path)
h_files.append(h_path)
outpath = os.path.join(plots_dir, f"{i}.png")
stack_images_vertically(h_files, outpath)
for v_file in h_files:
os.remove(v_file)
def save_infer_v_plot(infer_dir: str, sentence_ids: List[int]):
paths = []
for x in sentence_ids:
paths.append(os.path.join(infer_dir, f"{x}.png"))
path = os.path.join(infer_dir, f"{get_parent_dirname(infer_dir)}_v.png")
stack_images_vertically(paths, path)
def _save_comparison(dest_dir: str, dest_name: str, paths: List[str]) -> str:
path = os.path.join(dest_dir, f"{dest_name}.png")
stack_images_vertically(paths, path)
return path