train YOLOv5-AF-S

yolo/config/yolov5_af_config.py

@@ -54,7 +54,7 @@ class Yolov5AFBaseConfig(object):
         self.val_conf_thresh = 0.001
         self.val_nms_thresh  = 0.7
         self.test_topk = 100
-        self.test_conf_thresh = 0.3
+        self.test_conf_thresh = 0.4
         self.test_nms_thresh  = 0.5
 
         # ---------------- Assignment config ----------------

yolo/config/yolov5_config.py

@@ -57,7 +57,7 @@ class Yolov5BaseConfig(object):
         self.val_conf_thresh = 0.001
         self.val_nms_thresh  = 0.7
         self.test_topk = 100
-        self.test_conf_thresh = 0.3
+        self.test_conf_thresh = 0.4
         self.test_nms_thresh  = 0.5
 
         # ---------------- Assignment config ----------------