Recrosoftware/sd-webui-reactor
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Delete cimage.py

Browse Source
This commit is contained in:
Art Gourieff 2023-06-28 19:53:11 +07:00 committed by GitHub
parent 20e5bb5583
commit 4cb73f597a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 0 additions and 25 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
scripts/cimage.py
View File

@ -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