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