Skip to content

Employing a fusion of UNet and ResNet architectures, the project endeavors to achieve multiclass semantic segmentation of sandstone images. Through deep learning techniques, it seeks to uncover microstructural features across various geological classifications.

Notifications You must be signed in to change notification settings

arpsn123/Multiclass_Segmentation_using_by_UNET_with_RESNET_as_Backbone

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

7 Commits
 
 
 
 

Repository files navigation

UNet with ResNet Backbone: A Dynamic Fusion for Semantic Segmentation

In the realm of semantic segmentation, where pixel-level understanding of images is paramount, the synergy between UNet and ResNet emerges as a compelling solution. UNet, recognized for its intricate localization abilities, and ResNet, celebrated for its depth and feature extraction prowess, come together in a fusion that amplifies their individual strengths. This brief introduction sets the stage for exploring the amalgamation of UNet with ResNet as its backbone, unveiling its capacity to excel across diverse segmentation tasks while retaining architectural elegance and efficiency.

Dataset

The Sandstone dataset is a benchmark dataset used in image segmentation tasks, particularly in geological image analysis. It contains high-resolution images of sandstone rock samples, often obtained through microscopy techniques. Researchers use this dataset to develop and evaluate algorithms for segmenting different components within the sandstone samples, aiding in the understanding of their microstructure and properties.

This Dataset is annotated into 4 different classes :

  1. Bentheim (B49): Known for its uniformity and fine-grained structure, Bentheim sandstone is commonly used in construction and architectural applications due to its durability and aesthetic appeal.

  2. Berea (Br46): Berea sandstone is characterized by its high porosity and permeability, making it valuable for research in petroleum engineering and hydrology, as well as for geological studies.

  3. Fontainebleau (F57): Fontainebleau sandstone is renowned for its tight grain structure and exceptional strength, making it a popular choice for rock climbing holds and as a reference material in laboratory experiments.

  4. Gildehausen (G44): Gildehausen sandstone is notable for its heterogeneous composition, often exhibiting variations in grain size and mineral content, making it an interesting subject for geological research and analysis.

This Dataset originally contain 2 Files :

  1. images.tif
  2. masks.tif

Each being a "tiff stack file" having 1600 slices and dimension being 128x128, all the individual slices are extracted into 1600 single .tif files each having dimension : 128x128.

images_0 masks

Imgaes used for Training the model : 90% --> 1,440 images

Imgaes used for Testing the model: 10% --> 160 images

Model

Here, the segmentation_models library in Python, used extensively this for 2D image segmentation task. This module offers a streamlined approach to implementing the Pretrained Machine Learning Models, with the UNet model enhanced by ResNet serving as its backbone.

About

Employing a fusion of UNet and ResNet architectures, the project endeavors to achieve multiclass semantic segmentation of sandstone images. Through deep learning techniques, it seeks to uncover microstructural features across various geological classifications.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages