mirror of
https://github.com/vladmandic/automatic
synced 2026-09-19 01:04:32 +02:00
major refactoring of modules
Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
@@ -0,0 +1,21 @@
|
||||
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
|
||||
Reference in New Issue
Block a user