Prediction in Caffe - Exception: Input blob arguments do not match net inputs

后端 未结 4 1101
忘了有多久
忘了有多久 2021-02-14 02:02

I\'m using Caffe for classifying non-image data using a quite simple CNN structure. I\'ve had no problems training my network on my HDF5-data with dimensions n x 1 x 156 x 12. H

4条回答
  •  野的像风
    2021-02-14 02:42

    I have exactly the same problem. This is what fixed it.

    First, take the same prototext file as you used to train, remove the two data layers.

    Then add the block as Mark's above

    name: "Name_of_your_net"
    input: "data"
    input_dim: 64 
    input_dim: 1
    input_dim: 28
    input_dim: 28
    

    where my input_dim are for mnist, change them to your dim.

    Everything works.

提交回复
热议问题