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

Fix typos (#476)

Browse Source 
Found via `typos --format brief`
This commit is contained in:
Kian-Meng Ang 2023-06-22 15:55:11 +08:00 committed by GitHub
parent 8ce0b8f9ac
commit 8c02c27f21
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
README.md
View File

@ -312,7 +312,7 @@ The available models are:
### How to train your own model ### How to train your own model
If You need more fine tunned models try this: If You need more fine tuned models try this:
https://github.com/danielgatis/rembg/issues/193#issuecomment-1055534289 https://github.com/danielgatis/rembg/issues/193#issuecomment-1055534289

12
rembg/sessions/u2net_cloth_seg.py
View File

@ -9,7 +9,7 @@ from scipy.special import log_softmax
from .base import BaseSession from .base import BaseSession
pallete1 = [ palette1 = [
0, 0,
0, 0,
0, 0,
@ -24,7 +24,7 @@ pallete1 = [
0, 0,
] ]
pallete2 = [ palette2 = [
0, 0,
0, 0,
0, 0,
@ -39,7 +39,7 @@ pallete2 = [
0, 0,
] ]
pallete3 = [ palette3 = [
0, 0,
0, 0,
0, 0,
@ -76,17 +76,17 @@ class Unet2ClothSession(BaseSession):
masks = [] masks = []
mask1 = mask.copy() mask1 = mask.copy()
mask1.putpalette(pallete1) mask1.putpalette(palette1)
mask1 = mask1.convert("RGB").convert("L") mask1 = mask1.convert("RGB").convert("L")
masks.append(mask1) masks.append(mask1)
mask2 = mask.copy() mask2 = mask.copy()
mask2.putpalette(pallete2) mask2.putpalette(palette2)
mask2 = mask2.convert("RGB").convert("L") mask2 = mask2.convert("RGB").convert("L")
masks.append(mask2) masks.append(mask2)
mask3 = mask.copy() mask3 = mask.copy()
mask3.putpalette(pallete3) mask3.putpalette(palette3)
mask3 = mask3.convert("RGB").convert("L") mask3 = mask3.convert("RGB").convert("L")
masks.append(mask3) masks.append(mask3)