mirror of
https://github.com/vladmandic/sdnext.git
synced 2026-01-29 05:02:09 +03:00
22 lines
571 B
Python
22 lines
571 B
Python
import numpy as np
|
|
import torch
|
|
import torchvision.transforms.functional as vF
|
|
import PIL
|
|
|
|
|
|
JPEG_QUALITY = 95
|
|
|
|
|
|
def preprocess(image, processor, **kwargs):
|
|
if isinstance(image, PIL.Image.Image):
|
|
pass
|
|
elif isinstance(image, np.ndarray):
|
|
image = PIL.Image.fromarray(image)
|
|
elif isinstance(image, torch.Tensor):
|
|
image = vF.to_pil_image(image)
|
|
else:
|
|
raise TypeError(f"Image must be of type PIL.Image, np.ndarray, or torch.Tensor, got {type(image)} instead.")
|
|
|
|
image = processor.preprocess(image, **kwargs)
|
|
return image
|