test_caffe.py 文件源码

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

项目:tensorflow-action-conditional-video-prediction 作者: williamd4112 项目源码 文件源码
def __init__(self, mean, weight, K, num_act, num_step=1, data_path='test'):
        self.K = K
        self.num_act = num_act
        self.num_step = num_step

        caffe.set_mode_gpu()
        caffe.set_device(0)

        test_net_file, net_proto = N.create_netfile(1, data_path, mean, K, K,
            1, num_act, num_step=self.num_step, mode='test')

        self.test_net = caffe.Net(test_net_file, caffe.TEST)
        self.test_net.copy_from(weight)
评论列表
文章目录


问题


面经


文章

微信
公众号

扫码关注公众号