AI/rembg
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/danielgatis/rembg synced 2025-08-15 20:45:56 +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 .env
.envrc .envrc
.python-version
.idea .idea
.pytest_cache .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: if isinstance(data, bytes) or force_return_bytes:
return_type = ReturnType.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): elif isinstance(data, PILImage):
return_type = ReturnType.PILLOW return_type = ReturnType.PILLOW
img = data img = cast(PILImage, data)
elif isinstance(data, np.ndarray): elif isinstance(data, np.ndarray):
return_type = ReturnType.NDARRAY return_type = ReturnType.NDARRAY
img = Image.fromarray(data) img = cast(PILImage, Image.fromarray(data))
else: else:
raise ValueError( raise ValueError(
"Input type {} is not supported. Try using force_return_bytes=True to force python bytes output".format( "Input type {} is not supported. Try using force_return_bytes=True to force python bytes output".format(