From 4cb73f597a76cc31291d35dbba6af0181e20d832 Mon Sep 17 00:00:00 2001 From: Art Gourieff <85128026+Gourieff@users.noreply.github.com> Date: Wed, 28 Jun 2023 19:53:11 +0700 Subject: [PATCH] Delete cimage.py --- scripts/cimage.py | 25 ------------------------- 1 file changed, 25 deletions(-) delete mode 100644 scripts/cimage.py diff --git a/scripts/cimage.py b/scripts/cimage.py deleted file mode 100644 index bd154e8..0000000 --- a/scripts/cimage.py +++ /dev/null @@ -1,25 +0,0 @@ -from typing import List, Union, Dict, Set, Tuple - -from transformers import AutoFeatureExtractor -import torch -from PIL import Image, ImageFilter -import numpy as np - -def numpy_to_pil(images: np.ndarray) -> List[Image.Image]: - if images.ndim == 3: - images = images[None, ...] - images = (images * 255).round().astype("uint8") - pil_images = [Image.fromarray(image) for image in images] - - return pil_images - - -def check_image(x_image: np.ndarray) -> Tuple[np.ndarray, List[bool]]: - return x_image, False - - -def check_batch(x: torch.Tensor) -> torch.Tensor: - x_samples_ddim_numpy = x.cpu().permute(0, 2, 3, 1).numpy() - x_checked_image = x_samples_ddim_numpy - x = torch.from_numpy(x_checked_image).permute(0, 3, 1, 2) - return x