orobix / retina-unet

Retina blood vessel segmentation with a convolutional neural network

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

ValueError: CorrMM images and kernel must have the same stack size

skiyDev opened this issue · comments

Hi,
After execute run_testing.py script, i see following message:
ValueError: CorrMM images and kernel must have the same stack size

Apply node that caused the error: CorrMM{half, (1, 1), (1, 1)}(InplaceDimShuffle{0,2,3,1}.0, Subtensor{::, ::, ::int64, ::int64}.0)
Toposort index: 60
Inputs types: [TensorType(float32, 4D), TensorType(float32, 4D)]
Inputs shapes: [(32L, 16L, 24L, 48L), (3L, 3L, 64L, 32L)]
Inputs strides: [(73728L, 96L, 4L, 1536L), (4L, 12L, -1152L, -36L)]
Inputs values: ['not shown', 'not shown']
Outputs clients: [[Elemwise{Composite{((i0 + i1) + Abs((i0 + i1)))}}[(0, 0)](CorrMM{half, (1, 1), (1, 1)}.0, InplaceDimShuffle{x,0,x,x}.0)]]

keras version is 2.0.8
keras.json
{
"epsilon": 1e-07,
"floatx": "float32",
"image_data_format": "channels_last",
"backend": "theano",
"image_dim_ordering": "th"
}
os win server 2008 r2
python 2.7

I use test_best_weights.h5 and DRIVE database.