Jadzix/uhcs-microstructure-cnn

UHCS Microstructure CNN Classifier

A CNN model for classifying ultra-high carbon steel (UHCS) microstructures from microscopy images.

Model Description

Trained on the UHCS Microstructure dataset (Kaggle). Classifies grayscale microscopy images into 4 classes:

• —spheroidite
• —network
• —pearlite
• —martensite

Architecture

• —3 convolutional blocks (16/32/64 filters)
• —MaxPooling after each block
• —Fully connected layers (16384 -> 256 -> 4)
• —Dropout (p=0.5)
• —Input size: 128x128 grayscale

Performance

Usage

Model was trained with PyTorch. To load:

import torch
model = MicrostructureCNN()
model.load_state_dict(torch.load("best_model.pth"))
model.eval()

Dataset

UHCS Microstructure dataset on Kaggle

Full Project

Full code and notebook available on GitHub.

license: mit language:

• —en ---