update tensorboard metric

5 years ago · 5de4e25d68
parent 224e9a1d0b
commit 5de4e25d68
1 changed files with 1 additions and 1 deletions
--- a/train.py
+++ b/train.py
@ -308,7 +308,7 @@ def train(hyp):
        # Tensorboard
        if tb_writer:
            tags = ['train/giou_loss', 'train/obj_loss', 'train/cls_loss',
-                    'metrics/precision', 'metrics/recall', 'metrics/mAP_0.5', 'metrics/F1',
+                    'metrics/precision', 'metrics/recall', 'metrics/mAP_0.5', 'metrics/mAP_0.5:0.95',
                    'val/giou_loss', 'val/obj_loss', 'val/cls_loss']
            for x, tag in zip(list(mloss[:-1]) + list(results), tags):
                tb_writer.add_scalar(tag, x, epoch)