最新版本yolox训练得到的train_log.txt文件,用tensorboard查看total_loss曲线,需要怎么修改yolox/core/trainer.py文件
时间: 2023-06-17 21:04:49 浏览: 67
要查看 YOLOX 最新版本的训练日志的 total_loss 曲线,可以按照以下步骤进行操作:
1. 安装 Tensorboard
如果还没有安装 Tensorboard,可以使用以下命令在终端中安装:
```
pip install tensorboard
```
2. 修改 yolox/core/trainer.py 文件
找到 `train_epoch` 函数,将以下代码:
```python
# log
if self.local_rank == 0:
self.logger.info(
f'Epoch[{self.epoch}/{self.max_epoch}] LR: {self.lr_scheduler.get_lr()[0]:.6f} '
f'Time: {time.time() - st_time:.1f}s Loss: {loss.item():.4f} '
f'reg_loss: {reg_loss.item():.4f} '
f'cls_loss: {cls_loss.item():.4f} '
f'cent_loss: {cent_loss.item():.4f} '
f'total_loss: {total_loss.item():.4f} '
)
```
修改为以下代码:
```python
# log
if self.local_rank == 0:
self.logger.info(
f'Epoch[{self.epoch}/{self.max_epoch}] LR: {self.lr_scheduler.get_lr()[0]:.6f} '
f'Time: {time.time() - st_time:.1f}s Loss: {loss.item():.4f} '
f'reg_loss: {reg_loss.item():.4f} '
f'cls_loss: {cls_loss.item():.4f} '
f'cent_loss: {cent_loss.item():.4f} '
f'total_loss: {total_loss.item():.4f} '
)
writer = self.train_loader.dataset.writer
global_step = self.epoch * len(self.train_loader) + i
writer.add_scalar('Train/total_loss', total_loss.item(), global_step=global_step)
```
这样会在训练过程中将 total_loss 添加到 Tensorboard 的训练日志中。
3. 运行 Tensorboard
在终端中运行以下命令,启动 Tensorboard:
```
tensorboard --logdir /path/to/train_log/
```
其中 `/path/to/train_log/` 是训练日志文件所在的目录。
4. 查看 total_loss 曲线
在浏览器中打开 `http://localhost:6006`,即可查看 Tensorboard,选择 `Scalars` 标签页,选择 `Train/total_loss`,即可查看 total_loss 曲线。