3D-Unet-Segmentation

Implementation of 3D-Unet on Keras for sandstone segmentation.

The repository contains two experiments:

1. with the attention mechanism:
   

2. and without it:
   

Download data from gdrive