Getting tighter segmentation results of Deeplab for small/underbalanced classes

Question

I have trained the official Tensorflow deeplab model (https://github.com/tensorflow/models/tree/master/research/deeplab) on the CelebAMask-HQ dataset (https://github.com/swi