SergeMv
commented
6 years ago The answers are as follows:
train_crop_size is a parameter passed to data loader, and if you implement your own data loader, then you take it into account.
input_shape=(None, 3, None, None) works fine
n_classes should be 2
I use images of 320x320 px with 2 classes (background and foreground) with no cropping. What should I set in the following variables:
train_crop_size if train_crop_size = None is OK or should I use train_crop_size = (320, 320) ?
input_shape (there are 2 places where it is set) Is this OK: input_shape=(None, 3, 320, 320) Or should I use input_shape=(None, 3, None, None)
n_classes Since I have 2 classes, but your code assumes that the result of get_void_labels() is non empty, then I return [255] from get_void_labels() (value of 255 is not used on my mask files) and thus increase n_classes by 1. So I set n_classes to 3 is that correct?