Browse Source

train YOLOv1 on VOC

yjh0410 2 years ago
parent
commit
ca7e188e89
1 changed files with 1 additions and 1 deletions
  1. 1 1
      config/yolov1_config.py

+ 1 - 1
config/yolov1_config.py

@@ -16,7 +16,7 @@ yolov1_cfg = {
     'neck_depthwise': False,
     # head
     'head': 'decoupled_head',
-    'head_act': 'silu',
+    'head_act': 'lrelu',
     'head_norm': 'BN',
     'num_cls_head': 2,
     'num_reg_head': 2,