浏览代码

Merge pull request #730 from serg123e/fix-dis-normalization

fix IS-Net normalization values
Daniel Gatis 6 月之前
父节点
当前提交
7536338641
共有 1 个文件被更改,包括 1 次插入1 次删除
  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, :, :]