I would like to plot the accuracy of test data and val data through tensorboard in yolov7.

[nakyoungan]

Select Topic Area

Question

Body

I would like to plot the f1score of test data and val data through tensorboard in yolov7.

1. Firstly, how can I connect tensorboard to yolov7?

• Through other searches, I found out that I can do it like below. Can I add this to "train.py"?

import torch
from torch.utils.tensorboard import SummaryWriter
writer = SummaryWriter()

2. Next, yolov7 draws recall and precision by default, but I want to draw f1score separately for train and val.

What should I do?

[dsbyprateekg]

Select Topic Area

Question

Body

I would like to plot the f1score of test data and val data through tensorboard in yolov7.

1. Firstly, how can I connect tensorboard to yolov7?

• Through other searches, I found out that I can do it like below. Can I add this to "train.py"?

import torch
from torch.utils.tensorboard import SummaryWriter
writer = SummaryWriter()

2. Next, yolov7 draws recall and precision by default, but I want to draw f1score separately for train and val.

What should I do?

Use wandb instead of tensorboard.