Jelajahi Sumber

train YOLOv1 on VOC

yjh0410 2 tahun lalu
induk
melakukan
ca7e188e89
1 mengubah file dengan 1 tambahan dan 1 penghapusan
  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,