Update dw_head.py

this can sovle issue strongwolf#13 (comment)
XDUwsk · Aug 22, 2022 · 80613a2 · 80613a2
1 parent 27f50e5
commit 80613a2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dw_head.py b/dw_head.py
@@ -123,7 +123,7 @@ def forward_single(self, x, scale, stride):
         centerness = self.conv_centerness(reg_feat)
         bbox_pred = scale(bbox_pred).float()
         bbox_pred = F.relu(bbox_pred)
-        bbox_pred *= stride
+        bbox_pred = bbox_pred * stride
         if self.with_reg_refine:
             reg_dist = bbox_pred.permute(0, 2, 3, 1).reshape(-1, 4)
             points = self.prior_generator.single_level_grid_priors((h,w), self.strides.index(stride), dtype=x.dtype, device=x.device)