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

chore: Refactor remove function to handle unsupported input types

Browse Source
This commit is contained in:
Daniel Gatis 2024-08-04 09:55:19 -03:00
parent 90536b6858
commit 688b34841b
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
rembg/bg.py
View File

@ -241,7 +241,7 @@ def remove(
"""
if isinstance(data, bytes) or force_return_bytes:
return_type = ReturnType.BYTES
img = Image.open(io.BytesIO(data))
img = Image.open(io.BytesIO(cast(bytes, data)))
elif isinstance(data, PILImage):
return_type = ReturnType.PILLOW
img = data
@ -249,7 +249,11 @@ def remove(
return_type = ReturnType.NDARRAY
img = Image.fromarray(data)
else:
raise ValueError("Input type {} is not supported. Try using force_return_bytes=True to force python bytes output".format(type(data)))
raise ValueError(
"Input type {} is not supported. Try using force_return_bytes=True to force python bytes output".format(
type(data)
)
)
putalpha = kwargs.pop("putalpha", False)