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

Revert "rembg b - return a stream of RGB24 instead of a mess of consecutive pngs"

Browse Source
This commit is contained in:
calledit 2025-01-30 13:54:16 +01:00 committed by GitHub
parent 1f98d03b95
commit b9634aaf7e
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
rembg/commands/b_command.py
View File

@ -135,7 +135,9 @@ def b_command(
os.makedirs(output_dir, exist_ok=True) os.makedirs(output_dir, exist_ok=True)
def img_to_byte_array(img: PILImage) -> bytes: def img_to_byte_array(img: PILImage) -> bytes:
return img.convert('RGB').tobytes("raw", "RGB") buff = io.BytesIO()
img.save(buff, format="PNG")
return buff.getvalue()
async def connect_stdin_stdout(): async def connect_stdin_stdout():
loop = asyncio.get_event_loop() loop = asyncio.get_event_loop()