0

我想在用pytorch编写的 YOLOV5上训练时将所有训练指标记录到 csv 文件中,但问题是我不想使用 tensorboard。

为了实现这个目标,我尝试了如下一些技术:

-首先将其记录到tensorboard中,然后尝试将其转换为csv文件(失败)

- 从权重和偏差中提取日志文件(失败)

- 训练时直接写入csv文件(失败)

在这里,我尝试打开一个 csv 文件并将张量数据转换为字符串,但未能成功。

所以我真的很想知道我怎样才能实现这个目标,因为 YOLOV5 没有使用像 model.fit 这样的 pytorch 实用程序,我们可以使用回调函数。

提前致谢。

4

1 回答 1

1

您可以尝试使用权重和偏差 YOLOv5 集成。这是链接:https ://docs.wandb.ai/guides/integrations/yolov5

该链接有更多详细信息,但这里有一些表达基本思想的引文:

只需安装 wandb,您将激活内置的 W&B 日志记录功能:系统指标、模型指标和记录到交互式仪表板的媒体。

pip install wandb
git clone https://github.com/ultralytics/yolov5.git
python yolov5/train.py  # train a small network on a small dataset

只需按照 wandb.pip install wandb git clone 打印到标准的链接

您还可以进行模型版本控制和数据集可视化,这在 w&b yolov5 集成文档页面(上面链接)中有更多解释。您还可以观看此 YouTube 视频以获取指南:https ://youtu.be/yyecuhBmLxE

于 2021-12-17T17:21:17.973 回答