convolutional_pose_machine.py 文件源码

def __init__(self, n_point, n_stage):
        super(CPM, self).__init__()
        self.add_link('branch', Branch())
        self.add_link('stage1', Stage1(n_point))
        links = []
        for i in xrange(n_stage-1):
            links += [('stage{}'.format(i+2), StageN(n_point))]
        for l in links:
            self.add_link(*l)

        self.forward = links
        self.train = True