CornerNet: Detecting Objects as Paired Keypoints
PDF: https://arxiv.org/pdf/1808.01244.pdf
PyTorch代碼: https://github.com/shanglianlm0525/PyTorch-Networks
創新點:
- 將一個目標物體檢測爲一對關鍵點——邊界框的左上角和右下角。
- 使用單個卷積網絡來預測同一物體類別的所有實例的左上角和右下角的熱力圖,以及每個檢測到的角點的嵌入向量(用於對角點分組)。
- 提出corner pooling層,用於卷積網絡更好地定位邊界框的角點。
網絡結構:
corner pooling層的計算方式:
預測模塊結構圖
PyTorch代碼: