绿色为ASFF_Head
,浅蓝色Decoupled_Head
,深蓝色是第三步加的_initialize_dh_biases
方法后的效果。
参数量与计算量对比
模型 | 参数量 parameters | 计算量GFLOPs |
---|---|---|
yolov5s_Head | 7235389 | 16.5 |
ASFF_Head | 12674847 | 25.0 |
Decoupled_Head | 8928697 | 22.0 |
结构图
本篇介绍的这个Decouple_Head
和YOLOX
的头结构几乎相同,但这次的添加方式和我蓝皮书介绍的那篇是不同的,而且这个头优化的比较好,所以参数量和计算量都下降了,效果有可能和YOLOX
的头有差距,。