

绿色为
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的头有差距,。
转载:https://blog.csdn.net/weixin_43694096/article/details/129190682
查看评论