Skip to content

Commit

Permalink
Add warning emoji ⚠️ on --conf > 0.001 (ultralytics#8005)
Browse files Browse the repository at this point in the history
* Add warning emoji on `--conf > 0.001`

* Update val.py
  • Loading branch information
glenn-jocher authored and tdhooghe committed Jun 10, 2022
1 parent a356906 commit a259f39
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions val.py
Original file line number Diff line number Diff line change
Expand Up @@ -38,7 +38,7 @@
from utils.callbacks import Callbacks
from utils.dataloaders import create_dataloader
from utils.general import (LOGGER, check_dataset, check_img_size, check_requirements, check_yaml,
coco80_to_coco91_class, colorstr, increment_path, non_max_suppression, print_args,
coco80_to_coco91_class, colorstr, emojis, increment_path, non_max_suppression, print_args,
scale_coords, xywh2xyxy, xyxy2xywh)
from utils.metrics import ConfusionMatrix, ap_per_class, box_iou
from utils.plots import output_to_target, plot_images, plot_val_study
Expand Down Expand Up @@ -368,7 +368,7 @@ def main(opt):

if opt.task in ('train', 'val', 'test'): # run normally
if opt.conf_thres > 0.001: # https://github.com/ultralytics/yolov5/issues/1466
LOGGER.info(f'WARNING: confidence threshold {opt.conf_thres} >> 0.001 will produce invalid mAP values.')
LOGGER.info(emojis(f'WARNING: confidence threshold {opt.conf_thres} > 0.001 produces invalid results ⚠️'))
run(**vars(opt))

else:
Expand Down

0 comments on commit a259f39

Please sign in to comment.