Fixes a bug when using non-default image sizes

[askorelas]

Came across a bug when using smaller picture sizes than 256. The function load_train_data method wasn't given the picture shape that the network was built with, which resulted in incompatible tensor shapes.

[Deeplearning20]

@askorelas Hello, I changed my picture to 256*256. But there is a problem, how did you solve it?Thanks very much！ ValueError: Cannot feed value of shape (0,) for Tensor 'real_A_and_B_images:0', which has shape '(?, 256, 256, 6)'

[star0w]

@askorelas Hello, I changed my picture to 256*256. But there is a problem, how did you solve it?Thanks very much！ ValueError: Cannot feed value of shape (0,) for Tensor 'real_A_and_B_images:0', which has shape '(?, 256, 256, 6)'

hello! have u solved this problem?