zjc
/
yolov5
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Update COCO train postprocessing (#1702)

This commit is contained in:
Glenn Jocher 2020-12-15 21:50:28 -08:00 committed by GitHub
parent 035ac82ed0
commit 7947c86b57
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 12 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
train.py
View File

@ -402,15 +402,18 @@ def train(hyp, opt, device, tb_writer=None, wandb=None):
# Test best.pt # Test best.pt
if opt.data.endswith('coco.yaml') and nc == 80: # if COCO if opt.data.endswith('coco.yaml') and nc == 80: # if COCO
results, _, _ = test.test(opt.data, for conf, iou, save_json in ([0.25, 0.45, False], [0.001, 0.65, True]): # speed, mAP tests
batch_size=total_batch_size, results, _, _ = test.test(opt.data,
imgsz=imgsz_test, batch_size=total_batch_size,
model=attempt_load(best if best.exists() else last, device).half(), imgsz=imgsz_test,
single_cls=opt.single_cls, conf_thres=conf,
dataloader=testloader, iou_thres=iou,
save_dir=save_dir, model=attempt_load(best if best.exists() else last, device).half(),
save_json=True, # use pycocotools single_cls=opt.single_cls,
plots=False) dataloader=testloader,
save_dir=save_dir,
save_json=save_json,
plots=False)
else: else:
dist.destroy_process_group() dist.destroy_process_group()