瀏覽代碼

train YOLOv1 on VOC

yjh0410 2 年之前
父節點
當前提交
ca7e188e89
共有 1 個文件被更改,包括 1 次插入1 次删除
  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,