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

Question

我想在 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 张图像” 请帮助这是非常令人沮丧..

Answer 1

不幸的是，为了清楚起见，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 中看到详细输出。