Просмотр исходного кода

Fix inconsistent NMS IoU value for COCO (#3934)

Evaluation of 'best' and 'last' models will use the same params as the evaluation during the training phase. 
This PR fixes https://github.com/ultralytics/yolov5/issues/3907
modifyDataloader
Eldar Kurtic GitHub 3 лет назад
Родитель
Сommit
e7888af94c
Не найден GPG ключ соответствующий данной подписи Идентификатор GPG ключа: 4AEE18F83AFDEB23
1 измененных файлов: 0 добавлений и 2 удалений
  1. +0
    -2
      train.py

+ 0
- 2
train.py Просмотреть файл

@@ -457,8 +457,6 @@ def train(hyp, # path/to/hyp.yaml or hyp dictionary
results, _, _ = test.run(data_dict,
batch_size=batch_size // WORLD_SIZE * 2,
imgsz=imgsz_test,
conf_thres=0.001,
iou_thres=0.7,
model=attempt_load(m, device).half(),
single_cls=single_cls,
dataloader=testloader,

Загрузка…
Отмена
Сохранить