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

Update base.py

Browse Source
This commit is contained in:
PENGUINLIONG 2025-04-24 09:01:18 +08:00 committed by GitHub
parent fdedf86e56
commit 112e76d3cd
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
rembg/sessions/base.py
View File

@ -41,7 +41,7 @@ class BaseSession:
im = img.convert("RGB").resize(size, Image.Resampling.LANCZOS)
im_ary = np.array(im)
im_ary = im_ary / np.max(im_ary)
im_ary = im_ary / max(np.max(im_ary), 1e-6)
tmpImg = np.zeros((im_ary.shape[0], im_ary.shape[1], 3))
tmpImg[:, :, 0] = (im_ary[:, :, 0] - mean[0]) / std[0]