YOLOv10模型改进-Neck改进-第75篇:YOLOv10改进策略【Neck】| FPN-FMF特征匹配融合
一、本文介绍
本文记录的是利用FMF(Feature Matching Fusion)改进YOLOv10的Neck部分,实现特征匹配的特征融合。
二、FMF模块介绍
2.1 设计出发点
传统特征融合方法没有考虑特征间的匹配关系,FMF通过特征匹配实现更精准的特征融合。
2.2 模块结构
FMF融合过程:
- 特征匹配:计算不同尺度特征间的匹配度
- 加权融合:根据匹配度加权融合特征
- 特征增强:对融合后的特征进行增强
三、FMF的实现代码
importtorchimporttorch.nnasnnclassFMFModule