ARTICLE DETAIL

资讯详情

深耕编程入门与网站建设的一线实战洞察。

061、YOLOv11改进-CARAFE内容感知上采样替换Neck上采样即插即用涨点实验

061、YOLOv11改进-CARAFE内容感知上采样替换Neck上采样即插即用涨点实验 061、YOLOv11改进-CARAFE内容感知上采样替换Neck上采样即插即用涨点实验从一次深夜调试说起上个月跑YOLOv11训练一批无人机航拍数据,小目标检测效果始终上不去。mAP卡在0.72左右,调了anchor、改了loss权重,甚至试了增加小目标数据增强,效果都微乎其微。凌晨两点盯着tensorboard曲线发呆,突然意识到一个问题——Neck部分的上采样方式太粗糙了。YOLOv11默认用的是最近邻插值上采样,简单粗暴,但信息损失严重。对于小目标检测,特征图从深层往浅层传递时,细节信息被均匀地“抹平”了。当时脑子里闪过一个念头:能不能让上采样过程学会“看”内容,而不是机械地复制像素?于是翻出了CARAFE(Content-Aware ReAssembly of FEatures)这个模块。说实话,之前在其他项目里用过,效果不错,但一直没在YOLOv11上试过。当晚改完代码,重新训练,mAP直接跳到0.78。今天就把这个改进方案完整写下来。CARAFE到底干了什么传统的上采样方法,比如最近邻、双线性插值,本质上都是基于固定规则的像素填充。它们不知道特征图里哪些区域是边缘、哪些是纹理、哪些是平坦区域。CARAFE的核心思想是:让上采样过程根据输入特征的内容动态生成上采样核。具体来说,CARAFE分成两步:第一步,内容感知核预测。输入特征图经过一个小型卷积网络,输出一组上采样核。每个位置的上采样核大小是k_up × k_up,核的权重由该位置及其邻域的特征内容决定。/
返回列表