glenn-jocher commited on
Commit
09ba6f6
1 Parent(s): 2373c50

Add warning emoji ⚠️ on `--conf > 0.001` (#8005)

Browse files

* Add warning emoji on `--conf > 0.001`

* Update val.py

Files changed (1) hide show
  1. val.py +2 -2
val.py CHANGED
@@ -38,7 +38,7 @@ from models.common import DetectMultiBackend
38
  from utils.callbacks import Callbacks
39
  from utils.dataloaders import create_dataloader
40
  from utils.general import (LOGGER, check_dataset, check_img_size, check_requirements, check_yaml,
41
- coco80_to_coco91_class, colorstr, increment_path, non_max_suppression, print_args,
42
  scale_coords, xywh2xyxy, xyxy2xywh)
43
  from utils.metrics import ConfusionMatrix, ap_per_class, box_iou
44
  from utils.plots import output_to_target, plot_images, plot_val_study
@@ -363,7 +363,7 @@ def main(opt):
363
 
364
  if opt.task in ('train', 'val', 'test'): # run normally
365
  if opt.conf_thres > 0.001: # https://github.com/ultralytics/yolov5/issues/1466
366
- LOGGER.info(f'WARNING: confidence threshold {opt.conf_thres} >> 0.001 will produce invalid mAP values.')
367
  run(**vars(opt))
368
 
369
  else:
 
38
  from utils.callbacks import Callbacks
39
  from utils.dataloaders import create_dataloader
40
  from utils.general import (LOGGER, check_dataset, check_img_size, check_requirements, check_yaml,
41
+ coco80_to_coco91_class, colorstr, emojis, increment_path, non_max_suppression, print_args,
42
  scale_coords, xywh2xyxy, xyxy2xywh)
43
  from utils.metrics import ConfusionMatrix, ap_per_class, box_iou
44
  from utils.plots import output_to_target, plot_images, plot_val_study
 
363
 
364
  if opt.task in ('train', 'val', 'test'): # run normally
365
  if opt.conf_thres > 0.001: # https://github.com/ultralytics/yolov5/issues/1466
366
+ LOGGER.info(emojis(f'WARNING: confidence threshold {opt.conf_thres} > 0.001 produces invalid results ⚠️'))
367
  run(**vars(opt))
368
 
369
  else: