index 50 is out of bounds for axis 0 with size 50

[tracob-github]

Thanks for sharing this excellent code.

When I run test_aenet.py, an error occurred.

After debugging, I found that the size of test_lbs and z in models.py are inconsistent. The size of test_lbs is (50, 256, 256, 1), but the size of z is (800, 64, 64, 1). As show in two pictures.

I would like to ask how to sovle it.

[lucasmansilla]

Hi @tracob-github,

AE-Net runs with 64x64 segmentations. What size of segmentations are you using?

[tracob-github]

Thank you!I know where the problem is. I try to reproduce your work, and the accuracy of test_aenet is about 0.97. Is it correct？

[Westbrone]

Hello, your code is written very well, but because there is no pre-training model of the autoencoder, when I was training the model, the loss remained at around 0.49 in the end. Is it because of the pre-training model?