TPH-YOLOv5:改良版YOLOv5基于变压器预测头
在无人机捕捉场景中进行物体检测是最近受欢迎的任务。由于无人机始终在不同的高度导航,物体的尺度变化剧烈,这给网络优化带来了负担。此外,高速低空飞行带来了对密集堆积的物体的运动模糊,这导致了物体区分的巨大挑战。为了解决上述两个问题,我们提出了TPH-YOLOv5。基于YOLOv5,我们添加了一个额外的预测头来检测不同尺度的物体。
下载地址
用户评论