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

Return a stream of RGB24 instead of a garbled mess of consecutive pngs

Browse Source
This commit is contained in:
calledit 2025-01-29 12:27:24 +01:00 committed by GitHub
parent 37bbf63500
commit dbf1c17610
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
rembg/commands/b_command.py
View File

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