Add mask dilation/erosion option for training