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。在大尺寸featur