Yolov4 darknet中yolov4.conv.137含义及训练相关问题咨询
yolov4.conv.137权重来源说明
yolov4.conv.137是YOLOv4官方在COCO数据集上预训练得到的主干网络权重,既不是PASCAL VOC也不是ImageNet数据集训练产出的。
这个权重是官方将完整YOLOv4模型在COCO数据集完成全量训练后,截取前137个卷积层的参数导出的,专门用于自定义数据集训练的初始化,相比ImageNet预训练的分类主干权重,更适配目标检测任务的特征提取需求。
训练命令的训练模式说明
你所使用的命令!./darknet detector train ./obj.data ./yolov4-obj.cfg ./yolov4.conv.137 -dont_show -map默认属于不冻结任何层、仅用预训练权重做初始化的全量微调,不属于冻结部分层的迁移学习。
Darknet框架的训练指令默认不会自动冻结任意层参数,只要你没有在yolov4-obj.cfg文件中手动为前137层添加stopbackward=1配置,训练过程中所有层的参数都会随梯度更新:前137层基于预训练权重继续调整,后续检测头部分从随机初始化状态开始训练。
如果需要实现冻结主干、仅训检测头的迁移学习,你需要手动修改配置文件,为前137个卷积层增加stopbackward=1参数即可。
内容的提问来源于stack exchange,提问作者F.M.
相关产品推荐
相关产品推荐

