Browse Source

fix DIS normalization values

Sergey Evstegneiev 6 months ago
parent
commit
451c10d7dd
1 changed files with 1 additions and 1 deletions
  1. 1 1
      rembg/sessions/dis_general_use.py

+ 1 - 1
rembg/sessions/dis_general_use.py

@@ -24,7 +24,7 @@ class DisSession(BaseSession):
         """
         ort_outs = self.inner_session.run(
             None,
-            self.normalize(img, (0.485, 0.456, 0.406), (1.0, 1.0, 1.0), (1024, 1024)),
+            self.normalize(img, (0.5, 0.5, 0.5), (1.0, 1.0, 1.0), (1024, 1024)),
         )
 
         pred = ort_outs[0][:, 0, :, :]