Get output from Lasagne (python deep neural network framework)

后端 未结 2 761
伪装坚强ぢ
伪装坚强ぢ 2021-01-24 05:23

I loaded the mnist_conv.py example from official github of Lasagne.

At the and, I would like to predict my own example. I saw that \"lasagne.layers.get_output()\" should

相关标签:
2条回答
  • 2021-01-24 05:50

    As written in your error message, the input is expected to be a 4D tensor, of shape (n_samples, n_channel, width, height). In the MNIST case, n_channels is 1, and width and height are 28.

    But you are inputting a 2D tensor, of shape (28, 28). You need to add new axes, which you can do with exampleChar = exampleChar[None, None, :, :]

    exampleChar = np.zeros(28, 28)
    print exampleChar.shape 
    exampleChar = exampleChar[None, None, :, :]
    print exampleChar.shape
    

    outputs

    (28, 28)
    (1, 1, 28, 28)
    

    Note: I think you can use np.newaxis instead of None to add an axis. And exampleChar = exampleChar[None, None] should work too.

    0 讨论(0)
  • 2021-01-24 06:02

    First you try pass a single "image" into your network, which so it has the dimension (256,256).

    But it need a list of 3 dimensional data i.e. images, which in theano is implemented as 4D tensor.

    I don't see your full code, how you intended to use lasagne's interface, but if your code is written properly, from what I saw so far, I think you should convert your (256,256) data first to a one single channel image like (1,256,256), then make a list from either use more (1,256,256) data passed in a list e.g. [(1,256,256), (1,256,256), (1,256,256)], or make a list from this single example like [(1,256,256)]. Former you get and then pass a (3,1,256,256), latter a (1,1,256,256) 4D tensor, which will be accepted by lasagne interface.

    0 讨论(0)
提交回复
热议问题