AI/rembg
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/danielgatis/rembg synced 2025-08-05 20:46:03 +08:00
Code Issues Packages Projects Releases Wiki Activity

refactoring

Browse Source
This commit is contained in:
Daniel Gatis 2023-04-20 21:50:30 -03:00
parent c27f4e20f3
commit 08a3bdc053
7 changed files with 8 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
rembg/sessions/dis.py
View File

@ -34,7 +34,7 @@ class DisSession(BaseSession):
fname = f"{cls.name()}.onnx"
pooch.retrieve(
"https://github.com/danielgatis/rembg/releases/download/v0.0.0/isnet-general-use.onnx",
f"md5:fc16ebd8b0c10d971d3513d564d01e29",
"md5:fc16ebd8b0c10d971d3513d564d01e29",
fname=fname,
path=cls.u2net_home(),
progressbar=True,

4
rembg/sessions/sam.py
View File

@ -141,7 +141,7 @@ class SamSession(BaseSession):
pooch.retrieve(
"https://github.com/danielgatis/rembg/releases/download/v0.0.0/vit_b-encoder-quant.onnx",
f"md5:13d97c5c79ab13ef86d67cbde5f1b250",
"md5:13d97c5c79ab13ef86d67cbde5f1b250",
fname=fname_encoder,
path=cls.u2net_home(),
progressbar=True,
@ -149,7 +149,7 @@ class SamSession(BaseSession):
pooch.retrieve(
"https://github.com/danielgatis/rembg/releases/download/v0.0.0/vit_b-decoder-quant.onnx",
f"md5:fa3d1c36a3187d3de1c8deebf33dd127",
"md5:fa3d1c36a3187d3de1c8deebf33dd127",
fname=fname_decoder,
path=cls.u2net_home(),
progressbar=True,

2
rembg/sessions/silueta.py
View File

@ -36,7 +36,7 @@ class SiluetaSession(BaseSession):
fname = f"{cls.name()}.onnx"
pooch.retrieve(
"https://github.com/danielgatis/rembg/releases/download/v0.0.0/silueta.onnx",
f"md5:55e59e0d8062d2f5d013f4725ee84782",
"md5:55e59e0d8062d2f5d013f4725ee84782",
fname=fname,
path=cls.u2net_home(),
progressbar=True,

2
rembg/sessions/u2net.py
View File

@ -36,7 +36,7 @@ class U2netSession(BaseSession):
fname = f"{cls.name()}.onnx"
pooch.retrieve(
"https://github.com/danielgatis/rembg/releases/download/v0.0.0/u2net.onnx",
f"md5:60024c5c889badc19c04ad937298a77b",
"md5:60024c5c889badc19c04ad937298a77b",
fname=fname,
path=cls.u2net_home(),
progressbar=True,

2
rembg/sessions/u2net_cloth_seg.py
View File

@ -97,7 +97,7 @@ class Unet2ClothSession(BaseSession):
fname = f"{cls.name()}.onnx"
pooch.retrieve(
"https://github.com/danielgatis/rembg/releases/download/v0.0.0/u2net_cloth_seg.onnx",
f"md5:2434d1f3cb744e0e49386c906e5a08bb",
"md5:2434d1f3cb744e0e49386c906e5a08bb",
fname=fname,
path=cls.u2net_home(),
progressbar=True,

2
rembg/sessions/u2net_human_seg.py
View File

@ -36,7 +36,7 @@ class U2netHumanSegSession(BaseSession):
fname = f"{cls.name()}.onnx"
pooch.retrieve(
"https://github.com/danielgatis/rembg/releases/download/v0.0.0/u2net_human_seg.onnx",
f"md5:c09ddc2e0104f800e3e1bb4652583d1f",
"md5:c09ddc2e0104f800e3e1bb4652583d1f",
fname=fname,
path=cls.u2net_home(),
progressbar=True,

2
rembg/sessions/u2netp.py
View File

@ -36,7 +36,7 @@ class U2netpSession(BaseSession):
fname = f"{cls.name()}.onnx"
pooch.retrieve(
"https://github.com/danielgatis/rembg/releases/download/v0.0.0/u2netp.onnx",
f"md5:8e83ca70e441ab06c318d82300c84806",
"md5:8e83ca70e441ab06c318d82300c84806",
fname=fname,
path=cls.u2net_home(),
progressbar=True,