Explorar o código

debug RTCDet-v2

yjh0410 %!s(int64=2) %!d(string=hai) anos
pai
achega
6569be7f95
Modificáronse 1 ficheiros con 1 adicións e 1 borrados
  1. 1 1
      models/detectors/rtcdet_v2/rtcdet_v2_head.py

+ 1 - 1
models/detectors/rtcdet_v2/rtcdet_v2_head.py

@@ -76,7 +76,7 @@ class MultiLevelHead(nn.Module):
             [SingleLevelHead(
                 in_dims[level],
                 max(out_dim, num_classes),   # cls head dim
-                max(out_dim//4, 4*reg_max),  # reg head dim
+                max(out_dim, 4*reg_max),  # reg head dim
                 cfg['num_cls_head'],
                 cfg['num_reg_head'],
                 cfg['head_act'],