Image segmentation using U-Net