YOLOv4传输学习/微调

Question

我使用我的原始数据集和自定义yolov4配置文件训练了一个自定义yolov4配置文件模型，我将它称为“基本”YOLOv4模型。

现在，我想使用我创建的这个基本模型来重新训练这个模型，使用我手动增强的图像。我正在尝试重新培训我的模型，以尝试增加mAP和AP。所以我想用我的基本模型中的权重来训练一个新的yolov4模型，用手工增强的图像。

我在YOLOv4维基页面上看到，使用stopbackward = 1会冻结这些层，因此不会更新这些层中的权重，但是这会降低准确性。另外，我在./darknet partial cfg/yolov4.cfg yolov4.weights yolov4.conv.137 137 取出前137层那里读到了另一条信息。这是否意味着前137层被冻结在网络中，还是意味着您只是在对137层进行培训？

我的问题是：

1. 哪些代码实际上冻结了层，这样我就可以在我创建的基本YOLOv4模型上进行传输学习了？
2. 您建议冻结哪些层，在网络中第一个YOLO层之前的第一个137层？

提前谢谢你！

Answer 1

回答你的问题：

1. 如果你想使用转移学习，你不需要冻结任何层。你应该用你第一次跑步时所储存的重量开始训练。因此，您可以不使用darknet.exe detector train data/obj.data yolo-obj.cfg yolov4.conv.137来运行darknet.exe detector train data/obj.data yolo-obj.cfg backup/your_weights_file。权重存储在备份文件夹build\darknet\x64\backup\中。例如，命令可以如下所示：darknet.exe detector train data/obj.data yolo-obj.cfg backup/yolov4_2000.weights
2. 冻结层可以节省训练时间。什么是一个好的解决方案是首先训练模型与第一层冻结，然后解冻层，以完善您的学习。我不知道什么是一个好数量的层冻结在第一次运行，也许你可以测试它与试错。

Answer 2

命令“./深色网部分cfg/yolov4.cfg yolov4.权重yolov4.con.137”将“yolov4.ww”中前137层的权重转储到文件“yolov4.con.137”中，与培训无关。