Explorar el Código

debug RTCDet-M

yjh0410 hace 2 años
padre
commit
82a17ad0de
Se han modificado 1 ficheros con 2 adiciones y 2 borrados
  1. 2 2
      config/model_config/rtcdet_config.py

+ 2 - 2
config/model_config/rtcdet_config.py

@@ -305,7 +305,7 @@ rtcdet_cfg = {
         ## Neck: PaFPN
         'fpn': 'rtcdet_pafpn',
         'fpn_reduce_layer': 'conv',
-        'fpn_downsample_layer': 'dsblock',
+        'fpn_downsample_layer': 'conv',
         'fpn_core_block': 'elan_block',
         'fpn_branch_depth': 3,
         'fpn_expand_ratio': 0.5,
@@ -361,7 +361,7 @@ rtcdet_cfg = {
         ## Neck: PaFPN
         'fpn': 'rtcdet_pafpn',
         'fpn_reduce_layer': 'conv',
-        'fpn_downsample_layer': 'dsblock',
+        'fpn_downsample_layer': 'conv',
         'fpn_core_block': 'elan_block',
         'fpn_branch_depth': 3,
         'fpn_expand_ratio': 0.5,