Fix segment evolution keys (#9742)

* Update

* Cleanup
This commit is contained in:
Glenn Jocher authored and GitHub committed 2022-10-09 23:51:29 +02:00
1 parent 5ef69ef3e6
commit 209be932de
3 files changed
+6 -5

No files matched your search

+3 -1
View File
@@ -607,7 +607,9 @@ def main(opt, callbacks=Callbacks()):
results = train(hyp.copy(), opt, device, callbacks)
callbacks = Callbacks()
# Write mutation results
print_mutation(results, hyp.copy(), save_dir, opt.bucket)
keys = ('metrics/precision', 'metrics/recall', 'metrics/mAP_0.5', 'metrics/mAP_0.5:0.95', 'val/box_loss',
'val/obj_loss', 'val/cls_loss')
print_mutation(keys, results, hyp.copy(), save_dir, opt.bucket)
# Plot results
plot_evolve(evolve_csv)