Răsfoiți Sursa

train YOLOv1 on VOC

yjh0410 2 ani în urmă
părinte
comite
ca7e188e89
1 a modificat fișierele cu 1 adăugiri și 1 ștergeri
  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,