From 80613a2535cf60cb89c093b14a697ccf89e7af5c Mon Sep 17 00:00:00 2001
From: XDUwsk <40171372+XDUwsk@users.noreply.github.com>
Date: Mon, 22 Aug 2022 17:14:48 +0800
Subject: [PATCH] Update dw_head.py

this can sovle issue https://github.com/strongwolf/DW/issues/13#issue-1251466442
---
 dw_head.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/dw_head.py b/dw_head.py
index ea8b368..82eacfe 100644
--- 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)