@@ -20,7 +20,7 @@ def temporary_env_var(key: str, new_value: str | None):
|
||||
del os.environ[key]
|
||||
|
||||
|
||||
def images_to_pillow(images: Tensor | list[Tensor]) -> list[Image]:
|
||||
def images_to_pillow(images: Tensor | list[Tensor]) -> list[Image.Image]:
|
||||
pillow_images = []
|
||||
for _bn, image in enumerate(images):
|
||||
i = 255.0 * image.cpu().numpy()
|
||||
|
||||
Reference in New Issue
Block a user