szagoruyko / cvpr15deepcompare

Code and models for "Learning to Compare Image Patches via Convolutional Neural Networks"
http://imagine.enpc.fr/~zagoruys/deepcompare.html
Other
468 stars 169 forks source link

How to use the 2ch2stream network #18

Closed yuanxy92 closed 7 years ago

yuanxy92 commented 7 years ago

I want to test the 2ch2stream network, but the net:forward() function give the following error: image

the network I used is net = torch.load('./networks/2ch2stream/2ch2stream_yosemite.t7') My input tensor size is 2x64x64

Can anyone tell me where is wrong and how to use the network in correct way? Thanks very much!

yuanxy92 commented 7 years ago

I solved the problem already, the input size should be 1x2x64x64 Sorry, I am not familiar with Torch