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)