dnn应用的python代码_Python dnn.dnn_available方法代码示例

本文整理汇总了Python中theano.sandbox.cuda.dnn.dnn_available方法的典型用法代码示例。如果您正苦于以下问题：Python dnn.dnn_available方法的具体用法？Python dnn.dnn_available怎么用？Python dnn.dnn_available使用的例子？那么恭喜您, 这里精选的方法代码示例或许可以为您提供帮助。您也可以进一步了解该方法所在模块theano.sandbox.cuda.dnn的用法示例。

在下文中一共展示了dnn.dnn_available方法的19个代码示例，这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞，您的评价将有助于我们的系统推荐出更棒的Python代码示例。

示例1: test_old_pool_interface

​点赞 6

​

# 需要导入模块: from theano.sandbox.cuda import dnn [as 别名]

# 或者: from theano.sandbox.cuda.dnn import dnn_available [as 别名]

def test_old_pool_interface():

if not cuda.dnn.dnn_available():

raise SkipTest(cuda.dnn.dnn_available.msg)

testfile_dir = os.path.dirname(os.path.realpath(__file__))

fname = 'old_pool_interface.pkl'

with open(os.path.join(testfile_dir, fname), 'rb') as fp:

try:

pickle.load(fp)

except ImportError:

# Windows sometimes fail with nonsensical errors like:

# ImportError: No module named type

# ImportError: No module named copy_reg

# when "type" and "copy_reg" are builtin modules.

if sys.platform == 'win32':

exc_type, exc_value, exc_trace = sys.exc_info()

reraise(SkipTest, exc_value, exc_trace)

raise

开发者ID:muhanzhang，项目名称:D-VAE，代码行数:19，

示例2: test_dnn_conv_merge_mouts

​点赞 6

​

# 需要导入模块: from theano.sandbox.cuda import dnn [as 别名]

# 或者: from theano.sandbox.cuda.dnn import dnn_available [as 别名]

def test_dnn_conv_merge_mouts():

# make sure it doesn't attempt to output/alpha merge a convolution

# that has multiple clients.

if not cuda.dnn.dnn_available():

raise SkipTest(cuda.dnn.dnn_available.msg)

img = T.ftensor4()

kern = T.ftensor4()

out = T.ftensor4()

conv = dnn.dnn_conv(img, kern)

lr = numpy.asarray(0.05, dtype='float32')

if cuda.dnn.version() == -1:

# Can't merge alpha with cudnn v1

fr = conv + out

else:

fr = lr * (conv + out)

rr = conv * lr

f = theano.function([img, kern, out], [fr, rr], mode=mode_with_gpu)

convs = [n for n in f.maker.fgraph.toposort()

if isinstance(n.op, dnn.GpuDnnConv)]

assert len(convs) == 1

开发者ID:muhanzhang，项目名称:D-VAE，代码行数:26，

示例3: test_dnn_conv_merge_broad

​点赞 6

​

# 需要导入模块: from theano.sandbox.cuda import dnn [as 别名]

# 或者: from theano.sandbox.cuda.dnn import dnn_available [as 别名]

def test_dnn_conv_merge_broad():

# Make sure that we don't apply output_merge on broadcasted values.

if not cuda.dnn.dnn_available():

raise SkipTest(cuda.dnn.dnn_available.msg)

img = T.ftensor4()

kern = T.ftensor4()

conv = dnn.dnn_conv(img, kern)

lr = numpy.asarray(0.05, dtype='float32')

# this does broadcasting

fr = conv + lr

f = theano.function([img, kern], [fr])

convs = [n for n in f.maker.fgraph.toposort()

if isinstance(n.op, dnn.GpuDnnConv)]

assert len(convs) == 1

conv = convs[0]

# Assert output was not merged

assert isinstance(conv.inputs[2].owner.op, GpuAllocEmpty)

开发者ID:muhanzhang，项目名称:D-VAE，代码行数:23，

示例4: get_output_for

​点赞 6

​

# 需要导入模块: from theano.sandbox.cuda import dnn [as 别名]

# 或者: from theano.sandbox.cuda.dnn import dnn_available [as 别名]

def get_output_for(self, input, *args, **kwargs):

if not dnn_available:

raise RuntimeError("cudnn is not available.")

# by default we assume 'cross', consistent with earlier versions of conv2d.

conv_mode = 'conv' if self.flip_filters else 'cross'

# if 'border_mode' is one of 'valid' or 'full' use that.

# else use pad directly.

border_mode = self.border_mode if (self.border_mode is not None) else self.pad

conved = dnn.dnn_conv(img=input,

kerns=self.W,

subsample=self.strides,

border_mode=border_mode,

conv_mode=conv_mode

)

if self.b is None:

activation = conved

elif self.untie_biases:

activation = conved + self.b.dimshuffle('x', 0, 1, 2)

else:

activation = conved + self.b.dimshuffle('x', 0, 'x', 'x')

return self.nonlinearity(activation)

开发者ID:benanne，项目名称:kaggle-ndsb，代码行数:25，

示例5: setUp

​点赞 6

​

# 需要导入模块: from theano.sandbox.cuda import dnn [as 别名]

# 或者: from theano.sandbox.cuda.dnn import dnn_available [as 别名]

def setUp(self):

"""

Set up a test image and filter to re-use.

"""

skip_if_no_gpu()

if