fixed a bug.

c73a5814 · Tian Zhi · GitHub · 4697f876 · c73a5814
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

fcos_core/modeling/rpn/fcos/fcos.py fcos_core/modeling/rpn/fcos/fcos.py +1 -1

未找到文件。
--- a/fcos_core/modeling/rpn/fcos/fcos.py
+++ b/fcos_core/modeling/rpn/fcos/fcos.py
@@ -105,8 +105,8 @@ class FCOSHead(torch.nn.Module):

            bbox_pred = self.scales[l](self.bbox_pred(box_tower))
            if self.norm_reg_targets:
+                bbox_pred = F.relu(bbox_pred)
                if self.training:
-                    bbox_pred = F.relu(bbox_pred)
                    bbox_reg.append(bbox_pred)
                else:
                    bbox_reg.append(bbox_pred * self.fpn_strides[l])