能否使用预训练YOLO完成回归任务:图片浏览量预测相关技术问询
基于预训练
YOLO的单图浏览量回归任务方案 任务定义
- 任务类型:CNN监督回归任务
- 输入:单张图片
- 输出:图片对应的预测浏览次数
模型选型依据
- 选择
YOLO的核心逻辑:YOLO是基于COCO数据集完成预训练的目标检测模型,高浏览量图片通常包含的人脸、动物、文字等类别均在COCO数据集覆盖范围内,预训练权重可直接提取对浏览量预测有价值的特征 - 基线方案测试结果:已尝试采用权重冻结的预训练
VGGNet、MobileNet等CNN模型搭建回归任务,效果未达预期
训练约束
- 无足够算力支撑在十万级以上图片数据集上进行多轮训练,因此无法采用预训练模型微调方案
内容的提问来源于stack exchange,提问作者Rayleigh
相关产品推荐
相关产品推荐

