Tensorflow, Image Segmentation Convnet Invalidargumenterror: Input To Reshape Is A Tensor With 28800000 Values, But The Requested Shape Has 57600

I am trying to segment images from the BRATS challenge. I am using U-net in a combination of these two repositories: https://github.com/zsdonghao/u-net-brain-tumor https://github.

Solution 1:

You set your batch size as 1 in your tensorflow pipeline during training but feeding in 500 batch size in your testing data. Thats why the network requests only a tensor of shape 57600. You can either set your training batch size 500 or testing batch size as 1.