本篇文章是

北京航天航空大學

嘗試在

Yolov5的基礎上加入一些新的技巧tricks

,最終將

Transformer Prediction Head

加在Yolov5上,在

無人機小目標檢測

中取得了不錯的效果!

小目標神器!TPH-YOLOv5:將Transformer預測載入Yolov5!

論文連結:

https://

arxiv。org/pdf/2108。1153

9。pdf

摘要:

小目標神器!TPH-YOLOv5:將Transformer預測載入Yolov5!

小目標檢測實際情況

小目標神器!TPH-YOLOv5:將Transformer預測載入Yolov5!

演算法流程

小目標神器!TPH-YOLOv5:將Transformer預測載入Yolov5!

演算法結構

小目標神器!TPH-YOLOv5:將Transformer預測載入Yolov5!

Transformer Encoder

小目標神器!TPH-YOLOv5:將Transformer預測載入Yolov5!

CBAM結構

小目標神器!TPH-YOLOv5:將Transformer預測載入Yolov5!

混合矩陣

小目標神器!TPH-YOLOv5:將Transformer預測載入Yolov5!

實驗結果

小目標神器!TPH-YOLOv5:將Transformer預測載入Yolov5!

小目標神器!TPH-YOLOv5:將Transformer預測載入Yolov5!

檢驗效果圖

小目標神器!TPH-YOLOv5:將Transformer預測載入Yolov5!

效果確實很棒,值得學習

我是 @Sophia ,一起進步學習

往期論文推薦學習:

全新思路!阿里達摩院提出OadTR框架!將Transformer引入線上行為檢測!ICCV2021

高效新思路!例項分割QueryInst演算法!騰訊、華科聯合出品!入選ICCV 2021!

∞-former!最新變體Transformer!DeepMind 出品!

最新綜述!深度神經網路視覺識別!共計329篇文獻!西安交大、清華等合作出品!

超強YOLOP!CVPR 2022領域主席團隊開源!全景感知多工SOTA!

Fastformer:史上最強最快Transformer!清華、微軟亞洲研究院出品!

油畫渲染的新演算法 Paint Transformer!ICCV2021 Oral!

超強領先!Transformer影象復原效果顯著!

首次!阿里達摩院將Pure Transformer 應用於目標重識別ReID!效果顯著!

ResNet超強變體CoTNet!一種新穎的Transformer風格計算機視覺識別模組!京東AI開源!

新模型!Conformer!Transformer與CNN的超強融合!中科院、鵬程實驗室、華為出品

何愷明團隊出品!MoCo V3 自監督視覺Transformer

超強!MDETR:基於Transformer的端到端調製目標檢測神器!Facebook開源

新突破!Swin-UNet:基於純 Transformer 結構的醫學影象分割網路

秀!ImageNet又被Long-Short Transformer 霸榜!

新思路!ConViT:用“柔性”卷積歸納偏差改進視覺Transformer!

Transformer靠什麼得以闖入CV界秒殺CNN?

收藏!深入理解Transformer!

重新整理ImageNet最高分!谷歌提出Scaling Vision Transformers!

推薦 | Transformer最新成果!Learn to Dance with AIST++

進展 | 計算機視覺中的Transformer

超強總結!456篇參考文獻!《計算機視覺對抗攻擊與防禦》最新綜述!