目標檢測論文:Objects as Points及其PyTorch實現

Objects as Points
PDF: https://arxiv.org/pdf/1904.07850v1.pdf
PyTorch代碼: https://github.com/shanglianlm0525/PyTorch-Networks

創新點:
1 分配的錨點 (center point) 僅僅是放在位置上,沒有尺寸框。不需要手動設置閾值做前後景區分。
在這裏插入圖片描述

2 提取關鍵點特徵圖(keypoint heatmap)上的局部峯值點(local peaks)作爲錨點, 並且每個目標僅僅有一個正的錨點,因此不會用到NMS;

3 相比較其他目標檢測算法下采樣16倍尺度,本文僅僅下采樣4倍, 使用更大分辨率的輸出特徵圖,因此無需用到多重特徵圖錨點;

網絡結構圖
在這裏插入圖片描述
不同任務的網絡輸出:在這裏插入圖片描述

PyTorch代碼:


發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章