飞道的博客

改进YOLO系列 | 添加轻量化Decouple_Head 和 ASFF_Head

266人阅读  评论(0)


绿色为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_HeadYOLOX的头结构几乎相同,但这次的添加方式和我蓝皮书介绍的那篇是不同的,而且这个头优化的比较好,所以参数量和计算量都下降了,效果有可能和YOLOX的头有差距,。


转载:https://blog.csdn.net/weixin_43694096/article/details/129190682
查看评论
* 以上用户言论只代表其个人观点,不代表本网站的观点或立场