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

chore: Update .gitignore and add .python-version file

Browse Source
This commit is contained in:
Daniel Gatis 2024-08-04 10:28:08 -03:00
parent 688b34841b
commit ed1c29576f
3 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
.gitignore vendored
View File

@ -11,7 +11,6 @@ __pycache__/
*~≈
.env
.envrc
.python-version
.idea
.pytest_cache

1
.python-version Normal file
View File

@ -0,0 +1 @@
3.12.4

6
rembg/bg.py
View File

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