From 4ebeeca4050b0aa5ca165deebc7536abd722268b Mon Sep 17 00:00:00 2001 From: "Kin-Yiu, Wong" <102582011@cc.ncu.edu.tw> Date: Wed, 28 Sep 2022 16:58:35 +0800 Subject: [PATCH] mask code fix https://github.com/WongKinYiu/yolov7/issues/855 --- models/common.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/models/common.py b/models/common.py index 950288a..9cdbc34 100644 --- 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