Got following errors while running Densenet_MNIST.py ? about densenet-tensorflow HOT 2 CLOSED

This is because the image size is reduced each time it passes the layer.
At MNIST, I set nb_block = 2 so that the dense block and transition layer pass only two times.
(See ReadME)

The image size is halved from the transition layer (because, avg pooling)
It is also reduced by half in the first conv layer (7 * 7, stride = 2) (also max pooling too)

To summarize, the image size of MNIST is 28 * 28.
So,

28 -> 14(first conv) -> 6(max pool, The reason for not half is because of kernel size.) 
-> 3 (dense +transition) -> 1 (dense+transition) -> error !

Therefore, it is necessary to set the number of dense blocks according to the input image.

Thank you

from densenet-tensorflow.

I modified the code so that it can be run on MNIST.

from densenet-tensorflow.