绿色为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
的头有差距,。