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