我想在 Google colab 上训练 YOLOv3 Darknet 对象检测模型期间绘制 mAP 和损失图

如何解决我想在 Google colab 上训练 YOLOv3 Darknet 对象检测模型期间绘制 mAP 和损失图

我想在 Google colab 上训练 YOLOv3 暗网对象检测模型期间绘制 mAP 和损失图。我认为 google colab 没有 GUI，这就是它不显示任何图表的原因。

无论如何我都尝试过并使用此命令在训练期间绘制图形：

!darknet/darknet detector train custom_data/labelled_data.data darknet/cfg/yolov3_custom.cfg custom_weight/darknet53.conv.74 -map

也在训练结束时评估每个权重：

!darknet/darknet detector map custom_data/labelled_data.data darknet/cfg/yolov3_custom.cfg backup/yolov3_custom_2000.weights

并在另一次训练期间尝试过：

!/darknet/darknet detector train data/obj.data yolo-obj.cfg yolov4.conv.137 -dont_show -mjpeg_port 8090 -map

但是这些命令都没有用于绘制损失，也没有显示 mAP 图... 这是我在训练期间拥有的唯一信息： “1：640.424744，640.424744 平均，0.000000 率，3.913164 秒，64 张图像” 请帮助这是非常令人沮丧..

解决方法

不幸的是，为了清楚起见，Colab 将没有单独的窗口显示损失和 mAP 的进度。

要确保一切正常，请尝试以下操作：

1. 每当您在 Colab 中训练时添加 -dont_show，因为没有显示屏。

运行这个命令：

!./darknet detector train custom_data/labelled_data.data cfg/yolov3_custom.cfg custom_weight/darknet53.conv.74 -dont_show -map

2. 您可能错过了暗网文件夹下 chart.png 的位置。如果你在./content下编译了darknet，结构如下：

3. 将输出日志放入 txt 或日志文件：

   !./darknet 检测器 train custom_data/labelled_data.data cfg/yolov3_custom.cfg custom_weight/darknet53.conv.74 -dont_show -map |三通输出.log

在这种情况下，您将在暗网文件夹下的 output.log 中看到详细输出。