def _thread_save_image(tensor, base_dir, file_type: str = "jpg", normalize=False, prefix="", suffix="", base_num=0, placeholder="0", just_length=10): for idx, image in enumerate(tensor): if len(str(base_num + idx)) > just_length: raise RuntimeError( f"the length of {base_num + idx} > {just_length}") name = prefix + str(base_num + idx).rjust(just_length, placeholder)\ + suffix + "." + file_type name = _os.path.join(base_dir, name) _save_image(image, name, normalize=normalize)
def save_image(path: Path, img: torch.Tensor, name: str, idx: int) -> None: (path / "images").mkdir(exist_ok=True, parents=True) filename = path / "images" / f"{name}-{idx}.png" _save_image(img, filename)
def save_image(path: Path, x: torch.Tensor, name: str, idx: int) -> None: p = make_dir(path / "images" / str(idx)) filename = p / f"{name}-{idx}.png" if not filename.exists(): _save_image(x, filename)
def save_image(tensor, image_path, **kwargs): _save_image(tensor, image_path, **kwargs)