Hourglass模块 网络结构 代码

论文:https://arxiv.org/abs/1603.06937

代码:https://github.com/princeton-vl/pytorch_stacked_hourglass

结构图:


这个模块的代码为:




主要class Hourglass:这个类包含了图中的结构,为什么结构是这样的?代码中由递归,所以结果为如此?

一个小的组成部分Residual模块为如下图中左边的图:


Hourglass结构在多人姿态估计中使用,被引入到目标检测中,在CornerNet[1]和CenterNet[2]中两篇论文中使用Hourglass结构。


[1].Law H, Deng J. CornerNet: Detecting Objects as Paired Keypoints[C]. european conference on computer vision, 2018: 765-781

[2].Duan K, Bai S, Xie L, et al. Centernet: Keypoint triplets for object detection[C]//Proceedings of the IEEE International Conference on Computer Vision. 2019: 6569-6578.

你可能感兴趣的:(Hourglass模块 网络结构 代码)