semantic-segmentation-digital-rock-physics

KO | EN

gitlite — search

#python #java #python3 #arduino #golang #machine-learning #rust #html #flask #javascript #seismology #nodejs

semantic-segmentation-digital-rock-physics

★ 20 Open GitHub ↗

2D and 3D Unet, ResNet, ResUnet Architectures for Digital Rock Segmentation

// repository documentation

Was this content helpful?

★ 0 (0 ratings)

# Image segmentation in Digital Rock Physics

In this project, I have built and tested 2D and 3D UNet, U-ResNet, and ResUnet using Keras and 
Tensorflow implementation in Python and trained with datasets obtained from Professor 
Christoph Arns. Intersection over union (IOU) was used as a metric to see the performance of 
each model on each phase of the predicted image. For measuring the physical accuracy of the 
models, the Euler number for each phase was computed. The 2D UNet, U-ResNet, and ResUNet
are trained on 134 million pixels and 3D UNet, U-ResNet and ResUNet are trained on 37 million 
pixels. Due to parallel computing issues with the Tensorflow package, the image size to train on 
3D models was reduced. From multiple tests using different datasets, the models are working 
with about 99% accuracy on validation dataset with 2D while obtaining reasonable accuracy 
with 3D due to time and computing constraints.

Advisors: Masa Prodanovic & Christoph Arns

## Dataset and Workflow
![](https://raw.githubusercontent.com/AhadMomin/semantic-segmentation-digital-rock-physics/master/Workflow.png)
![](https://raw.githubusercontent.com/AhadMomin/semantic-segmentation-digital-rock-physics/master/Training_set.png)

## CNN Architectures

### 2D UNet
![](https://raw.githubusercontent.com/AhadMomin/semantic-segmentation-digital-rock-physics/master/2D_Unet.png)

### 3D UNet
![](https://raw.githubusercontent.com/AhadMomin/semantic-segmentation-digital-rock-physics/master/3D_Unet.png)

### 2D ResUNet
![](https://raw.githubusercontent.com/AhadMomin/semantic-segmentation-digital-rock-physics/master/2D_Unet.png)

### 3D ResUNet
![](https://raw.githubusercontent.com/AhadMomin/semantic-segmentation-digital-rock-physics/master/3D_ResUNet.png)

### 2D U-ResNet
![](https://raw.githubusercontent.com/AhadMomin/semantic-segmentation-digital-rock-physics/master/2D_U-ResNet.png)

### 3D U-ResNEt
![](https://raw.githubusercontent.com/AhadMomin/semantic-segmentation-digital-rock-physics/master/3D_U-ResNet.png)

### Paper link:
https://ahadmomin.github.io/files/Final%20Report-%20Ahad%20Momin.pdf

This workflow is adapted for segmentation software which will be used for Digital Rock Physics Research.
For 3D workflows please request it here. For more datasets, check out: https://www.digitalrocksportal.org/

semantic-segmentation-digital-rock-physics

Explore Similar Repositories

Was this content helpful?

Recent Feedback

Download README