소스 검색

Add `--conf-thres` >> 0.001 warning (#5567)

Partially addresses invalid mAPs at higher confidence threshold issue https://github.com/ultralytics/yolov5/issues/1466.
modifyDataloader
Glenn Jocher GitHub 2 년 전
부모
커밋
0de4a9c35d
No known key found for this signature in database GPG Key ID: 4AEE18F83AFDEB23
1개의 변경된 파일2개의 추가작업 그리고 0개의 파일을 삭제
  1. +2
    -0
      val.py

+ 2
- 0
val.py 파일 보기

@@ -330,6 +330,8 @@ def main(opt):
check_requirements(requirements=ROOT / 'requirements.txt', exclude=('tensorboard', 'thop'))

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.')
run(**vars(opt))

elif opt.task == 'speed': # speed benchmarks

Loading…
취소
저장