Downloads · 30 days
0
Breast_MNIST_CNN
J-Aamir/Breast_MNIST_CNN
Breast_MNIST_CNN is a machine learning model from J-Aamir. Use it for the machine learning task on the model card, and read the license before you ship it in a product.
This repository contains a fine-tuned ResNet18 model for binary breast ultrasound image classification using the BreastMNIST dataset from MedMNIST.
Access
Public
Updated Jul 20, 2026
Repo size
44.8 MB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.pth44.8 MB · 100%
At a glance
- Access
- Public
- Created
- Jul 20, 2026
- Updated
- Jul 20, 2026
- SHA
- b395e05f
Model card
From the Hugging Face model README
BreastMNIST ResNet18 with Grad-CAM
Overview
This repository contains a fine-tuned ResNet18 model for binary breast ultrasound image classification using the BreastMNIST dataset from MedMNIST.
The project demonstrates:
- Transfer Learning
- Medical Image Classification
- Explainable AI (Grad-CAM)
- Error Analysis
Dataset
BreastMNIST (MedMNIST v2)
Task: Binary Classification
Classes
- 0 → Benign
- 1 → Malignant
Model
Architecture:
- ResNet18
Framework:
- PyTorch
Input Size:
- 224×224
Output:
- Binary Classification
Explainability
Grad-CAM visualizations are included in the GitHub repository.
GitHub Repository
https://github.com/J-Aamir/BreastMNIST-ResNet18-GradCAM
Intended Use
Educational and research purposes only.
Not intended for clinical diagnosis.
- Created
- Jul 20, 2026
- Updated
- Jul 20, 2026