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

Merge pull request #719 from calledit/revert-718-patch-1

Browse Source 
Revert "rembg b - return a stream of RGB24 instead of a mess of consecutive pngs"
This commit is contained in:
Daniel Gatis 2025-01-30 11:29:03 -03:00 committed by GitHub
commit a34d440fca
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)
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():
loop = asyncio.get_event_loop()