unet-uff-tensorrt. Training code for UNet segmentation model which can be converted to TensorRT engine