Browse Source

refactoring

Daniel Gatis 2 năm trước cách đây
mục cha
commit
08a3bdc053

+ 1 - 1
rembg/sessions/dis.py

@@ -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,

+ 2 - 2
rembg/sessions/sam.py

@@ -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,

+ 1 - 1
rembg/sessions/silueta.py

@@ -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,

+ 1 - 1
rembg/sessions/u2net.py

@@ -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,

+ 1 - 1
rembg/sessions/u2net_cloth_seg.py

@@ -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,

+ 1 - 1
rembg/sessions/u2net_human_seg.py

@@ -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,

+ 1 - 1
rembg/sessions/u2netp.py

@@ -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,