glenn-jocher commited on
Commit
5de4e25
1 Parent(s): 224e9a1

update tensorboard metric

Browse files
Files changed (1) hide show
  1. train.py +1 -1
train.py CHANGED
@@ -308,7 +308,7 @@ def train(hyp):
308
  # Tensorboard
309
  if tb_writer:
310
  tags = ['train/giou_loss', 'train/obj_loss', 'train/cls_loss',
311
- 'metrics/precision', 'metrics/recall', 'metrics/mAP_0.5', 'metrics/F1',
312
  'val/giou_loss', 'val/obj_loss', 'val/cls_loss']
313
  for x, tag in zip(list(mloss[:-1]) + list(results), tags):
314
  tb_writer.add_scalar(tag, x, epoch)
 
308
  # Tensorboard
309
  if tb_writer:
310
  tags = ['train/giou_loss', 'train/obj_loss', 'train/cls_loss',
311
+ 'metrics/precision', 'metrics/recall', 'metrics/mAP_0.5', 'metrics/mAP_0.5:0.95',
312
  'val/giou_loss', 'val/obj_loss', 'val/cls_loss']
313
  for x, tag in zip(list(mloss[:-1]) + list(results), tags):
314
  tb_writer.add_scalar(tag, x, epoch)