Hourglass模块 网络结构 代码
2020-10-06 本文已影响0人
元水1314
论文: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.