← all repositories

nyukat/breast_cancer_classifier

A deep learning model that classifies breast cancer from screening mammography images, trained on mammography exams with four standard views.

891 stars Jupyter Notebook Computer VisionDomain Apps
breast_cancer_classifier
Velocity · 7d
+0.3
★ / day
Trend
steady
star history

The project provides pretrained deep neural network models for breast cancer screening from mammography images. It includes two model variants: an image-only model that takes mammography images as input, and an image-and-heatmaps model that additionally incorporates patch-level prediction heatmaps. The models process four standard mammography views (L-CC, R-CC, L-MLO, R-MLO) and output probability predictions for benign and malignant findings per breast. Implemented in PyTorch.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.