本文主要是介绍TPH-yolov5 小目标检测,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
原文链接:https://blog.csdn.net/stf1065716904/article/details/125487840
0. 序言
论文:《TPH-YOLOv5: Improved YOLOv5 Based on Transformer Prediction Head for Object Detection on Drone-captured Scenarios》
代码:GitHub - TfeiSong/tph-yolov5: CBAM-TPH-biFPN-yolov5
TPH:Transformer Prediction Head
总结比较实用的点:
-
tph(transformer prediction head)
。能够捕获更多的全局信息和上下文信息(使用的self-attention和全连接)。
-
CBAM
提高通道和空间注意力机制。简单有效好嵌入CNN架构。
-
增加小目标预测head
这篇关于TPH-yolov5 小目标检测的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!