目标检测论文: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代码:


发布了270 篇原创文章 · 获赞 344 · 访问量 65万+

猜你喜欢

转载自blog.csdn.net/shanglianlm/article/details/101040188
今日推荐