fix warning typo

furyhawk · Sep 2, 2022 · 130c870 · 130c870
1 parent 803afca
commit 130c870
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/yolov6/core/evaler.py b/yolov6/core/evaler.py
@@ -247,8 +247,8 @@ def check_thres(conf_thres, iou_thres, task):
         '''Check whether confidence and iou threshold are best for task val/speed'''
         if task != 'train':
             if task == 'val':
-                if conf_thres > 0.01:
-                    LOGGER.warning(f'The best conf_thresh when evaluate the model is less than 0.01, while you set it to: {conf_thres}')
+                if conf_thres > 0.03:
+                    LOGGER.warning(f'The best conf_thresh when evaluate the model is less than 0.03, while you set it to: {conf_thres}')
                 if iou_thres != 0.65:
                     LOGGER.warning(f'The best iou_thresh when evaluate the model is 0.65, while you set it to: {iou_thres}')
             if task == 'speed' and conf_thres < 0.25: