ICCVAug, 2021
TPH-YOLOv5:基于 Transformer 预测头的改进 YOLOv5 用于无人机捕获的场景物体检测
TPH-YOLOv5: Improved YOLOv5 Based on Transformer Prediction Head for Object Detection on Drone-captured Scenarios
Xingkui Zhu, Shuchang Lyu, Xu Wang, Qi Zhao
TL;DRTPH-YOLOv5 是一种基于 YOLOv5 的目标检测模型,包括添加预测头、采用自注意力机制的 Transformer 预测头和卷积块关注模型的集成,以及有效策略的利用。 在无人机捕获的场景中展现出了优秀的性能表现。