This repo is adapted from Implementation of deep learning framework -- Unet, using Keras. U-Net for Rock Segmentation and Analysis Training trainUnet.ipynb Results Raw photo, ground truth, prediction and heat map: