mmdetection使用tensorboard
时间: 2023-11-14 09:07:52 浏览: 225
mmdetection-to-tensorrt:将mmdetection模型转换为tensorrt,支持fp16,int8,批输入,动态形状等
mmdetection是一个基于PyTorch的开源目标检测框架,它支持使用tensorboard进行训练过程的可视化。如果你想要使用tensorboard,需要在mmdetection的配置文件中进行设置。具体步骤如下:
1. 打开mmdetection/configs/base/default_runtime.py文件。
2. 找到log_config字典,将interval设置为你想要的输出间隔。
3. 将dict(type='TextLoggerHook')注释掉,去掉dict(type='TensorboardLoggerHook')的注释。
4. 运行训练脚本,tensorboard日志将会被保存在指定的目录中。
阅读全文