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)
test_caffe.py 文件源码
python
阅读 16
收藏 0
点赞 0
评论 0
评论列表
文章目录