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

verbose on final_epoch (#1997)

This commit is contained in:
Glenn Jocher 2021-01-20 18:27:38 -08:00 committed by GitHub
parent d9212140b3
commit 046c37e465
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
test.py
View File

@ -37,7 +37,6 @@ def test(data,
plots=True, plots=True,
log_imgs=0, # number of logged images log_imgs=0, # number of logged images
compute_loss=None): compute_loss=None):
# Initialize/load model and set device # Initialize/load model and set device
training = model is not None training = model is not None
if training: # called by train.py if training: # called by train.py
@ -227,7 +226,7 @@ def test(data,
print(pf % ('all', seen, nt.sum(), mp, mr, map50, map)) print(pf % ('all', seen, nt.sum(), mp, mr, map50, map))
# Print results per class # Print results per class
if (verbose or (nc <= 20 and not training)) and nc > 1 and len(stats): if (verbose or (nc < 50 and not training)) and nc > 1 and len(stats):
for i, c in enumerate(ap_class): for i, c in enumerate(ap_class):
print(pf % (names[c], seen, nt[c], p[i], r[i], ap50[i], ap[i])) print(pf % (names[c], seen, nt[c], p[i], r[i], ap50[i], ap[i]))

1
train.py
View File

@ -344,6 +344,7 @@ def train(hyp, opt, device, tb_writer=None, wandb=None):
single_cls=opt.single_cls, single_cls=opt.single_cls,
dataloader=testloader, dataloader=testloader,
save_dir=save_dir, save_dir=save_dir,
verbose=nc < 50 and final_epoch,
plots=plots and final_epoch, plots=plots and final_epoch,
log_imgs=opt.log_imgs if wandb else 0, log_imgs=opt.log_imgs if wandb else 0,
compute_loss=compute_loss) compute_loss=compute_loss)