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.

上一篇 下一篇

猜你喜欢

热点阅读