Train network with custom image size

[jsmithdlc]

Hi,

Thanks for your work. I've used the script for training the model at the 2048 dataset image size and 352 input image size but I'm struggling to get it to work with sizes different than those.

Is it possible? The images from my dataset are 512 x 512 and lowres, so I'd like to experiment with input images smaller than 352 x 352.