Ankle/yolov5_birdproject
1
0 Fork 0
Code Issues Pull Requests Actions 3 Packages Projects Releases Wiki Activity

update tensorboard metric

Browse Source
This commit is contained in:
Glenn Jocher committed 2020-07-11 11:39:02 -07:00
1 parent 224e9a1d0b
commit 5de4e25d68
1 file changed
+1 -1
Download Patch File Download Diff File Expand all files Collapse all files
train.py
+1 -1
View File
@@ -308,7 +308,7 @@ def train(hyp):
# Tensorboard # Tensorboard
if tb_writer: if tb_writer:
tags = ['train/giou_loss', 'train/obj_loss', 'train/cls_loss', 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'] 'val/giou_loss', 'val/obj_loss', 'val/cls_loss']
for x, tag in zip(list(mloss[:-1]) + list(results), tags): for x, tag in zip(list(mloss[:-1]) + list(results), tags):
tb_writer.add_scalar(tag, x, epoch) tb_writer.add_scalar(tag, x, epoch)