论文:Kwang Moo Yi, Eduard Trulls, Vincent Lepetit, Pascal Fua, ” LIFT: Learned Invariant Feature Transform”, in ECCV 2016, https://arxiv.org/abs/1603.09114
LIFT github 代码: https://github.com/cvlab-epfl/LIFT
所需软件及其正确版本: CUDA (8.0), cuDNN (5.1), python (2.7), theano (0.90), Lasagne (0.2.dev1), flufl.lock (2.4.1), 剩下的numpy, scipy, parse, h5py 版本没有具体要求。同时需要强调的是,有要求的版本必须严格匹配。
一. 相关软件包安装
1.CUDA (8.0), cuDNN (5.1)
相关package在该链接可找到 (https://developer.nvidia.com/), 具体操作参考网上教程。
测试代码:
from theano import function, config, shared, sandbox
import theano.tensor as T
import numpy
import time
vlen = 10 * 30 * 768 # 10 x #cores x # threads per core
iters = 1000
rng = numpy.random.RandomState(22)
x = shared(nump