AI/rembg
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/danielgatis/rembg synced 2025-08-06 00:16:27 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #441 from Flippchen/fix_image_orientation

Browse Source 
added fix for image orientation
This commit is contained in:
Daniel Gatis 2023-05-07 14:06:00 -03:00 committed by GitHub
commit ec29de6ca6
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
rembg/bg.py
View File

@ -11,7 +11,7 @@ from cv2 import (
getStructuringElement,
morphologyEx,
)
from PIL import Image
from PIL import Image, ImageOps
from PIL.Image import Image as PILImage
from pymatting.alpha.estimate_alpha_cf import estimate_alpha_cf
from pymatting.foreground.estimate_foreground_ml import estimate_foreground_ml
@ -113,6 +113,10 @@ def apply_background_color(img: PILImage, color: Tuple[int, int, int, int]) -> P
return colored_image
def fix_image_orientation(img: PILImage) -> PILImage:
return ImageOps.exif_transpose(img)
def remove(
data: Union[bytes, PILImage, np.ndarray],
alpha_matting: bool = False,
@ -138,6 +142,9 @@ def remove(
else:
raise ValueError("Input type {} is not supported.".format(type(data)))
# Fix image orientation
img = fix_image_orientation(img)
if session is None:
session = new_session("u2net", *args, **kwargs)