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代码: