mask code

fix https://github.com/WongKinYiu/yolov7/issues/855
2022-09-28 16:58:35 +08:00 · 2022-09-28 16:58:35 +08:00 · 4ebeeca405
parent fb797838bb
commit 4ebeeca405
1 changed files with 1 additions and 1 deletions
--- a/models/common.py
+++ b/models/common.py
@ -472,7 +472,7 @@ class ImplicitA(nn.Module):
    

 class ImplicitM(nn.Module):
-    def __init__(self, channel, mean=0., std=.02):
+    def __init__(self, channel, mean=1., std=.02):
        super(ImplicitM, self).__init__()
        self.channel = channel
        self.mean = mean