fix PIL exif rotate

This commit is contained in:
Qing 2024-01-04 21:39:59 +08:00
parent 4a9f2ab03c
commit 2fc1bf042a
2 changed files with 28 additions and 3 deletions

View File

@ -309,11 +309,12 @@ def decode_base64_to_image(
image = Image.open(io.BytesIO(base64.b64decode(encoding)))
alpha_channel = None
infos = image.info
try:
image = ImageOps.exif_transpose(image)
except:
pass
# exif_transpose will remove exif rotate infowe must call image.info after exif_transpose
infos = image.info
if gray:
image = image.convert("L")
@ -352,6 +353,29 @@ def concat_alpha_channel(rgb_np_img, alpha_channel) -> np.ndarray:
return rgb_np_img
def adjust_mask(mask: np.ndarray, kernel_size: int, operate):
# kernel_size = kernel_size*2+1
mask[mask >= 127] = 255
mask[mask < 127] = 0
# fronted brush color "ffcc00bb"
if operate == "expand":
mask = cv2.dilate(
mask,
np.ones((kernel_size, kernel_size), np.uint8),
iterations=1,
)
else:
mask = cv2.erode(
mask,
np.ones((kernel_size, kernel_size), np.uint8),
iterations=1,
)
res_mask = np.zeros((mask.shape[0], mask.shape[1], 4), dtype=np.uint8)
res_mask[mask > 128] = [255, 203, 0, int(255 * 0.73)]
res_mask = cv2.cvtColor(res_mask, cv2.COLOR_BGRA2RGBA)
return res_mask
def gen_frontend_mask(bgr_or_gray_mask):
if len(bgr_or_gray_mask.shape) == 3 and bgr_or_gray_mask.shape[2] != 1:
bgr_or_gray_mask = cv2.cvtColor(bgr_or_gray_mask, cv2.COLOR_BGR2GRAY)

View File

@ -2,6 +2,7 @@ from typing import List, Dict
import torch
from loguru import logger
import numpy as np
from lama_cleaner.download import scan_models
from lama_cleaner.helper import switch_mps_device
@ -77,12 +78,12 @@ class ModelManager:
config:
Returns:
BGR image
"""
self.switch_controlnet_method(config)
self.enable_disable_freeu(config)
self.enable_disable_lcm_lora(config)
return self.model(image, mask, config)
return self.model(image, mask, config).astype(np.uint8)
def scan_models(self) -> List[ModelInfo]:
available_models = scan_models()