我可以将预训练的 YOLO 用于回归任务吗？答案

【问题标题】：Can I use pretrained YOLO for a regression task?我可以将预训练的 YOLO 用于回归任务吗？
【发布时间】：2021-11-07 22:17:51
【问题描述】：

我对迁移学习很陌生。我打算使用 YOLO 来完成 CNN 监督回归任务。给定一张图片，预测它被查看的次数。我倾向于使用 YOLO，因为它是一个物体检测器。高浏览量的照片大多包含对象（人脸、动物、文本等），这些对象是 COCO 数据集中最初训练 YOLO 的类。

我已经尝试使用具有冻结权重的预训练 CNN 模型（VGGNet、MobileNet 等），但结果并不好。微调预训练模型的选项是不可能的，因为我没有计算资源来训练 x epoch 的 100K+ 图像来为我的问题创建一个好的模型。

【问题讨论】：

【解决方案1】：

YOLO使用暗网作为 CNN 主干网/特征提取器。因此，您可能想尝试使用预训练的暗网作为特征提取器，并将分类器替换为您的回归器。 Standard YOLO 使用 Darknet-53，而 Tiny YOLO 使用 Darknet-19。

【讨论】：

这里是暗网网站的链接 - pjreddie.com/darknet/imagenet
我花了一些时间探索暗网架构。我一直在创建数据文件（./darknet classifier train ... 所需的参数），该文件预计包含classes 和labels。我没有的东西，因为我打算做回归。你有什么建议可以帮助我吗？非常感谢。