目標檢測論文:CornerNet: Detecting Objects as Paired Keypoints及其PyTorch實現

CornerNet: Detecting Objects as Paired Keypoints
PDF: https://arxiv.org/pdf/1808.01244.pdf
PyTorch代碼: https://github.com/shanglianlm0525/PyTorch-Networks
創新點:

  1. 將一個目標物體檢測爲一對關鍵點——邊界框的左上角和右下角。
  2. 使用單個卷積網絡來預測同一物體類別的所有實例的左上角和右下角的熱力圖,以及每個檢測到的角點的嵌入向量(用於對角點分組)。
  3. 提出corner pooling層,用於卷積網絡更好地定位邊界框的角點。

網絡結構:
在這裏插入圖片描述
corner pooling層的計算方式:在這裏插入圖片描述
預測模塊結構圖
在這裏插入圖片描述
PyTorch代碼:


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