U-Net for defect segmentation