__init__.py 文件源码

python
阅读 20 收藏 0 点赞 0 评论 0

项目:convolutional-pose-machines-chainer 作者: tomoyukun 项目源码 文件源码
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
评论列表
文章目录


问题


面经


文章

微信
公众号

扫码关注公众号