The Breast Cancer Screening Framework

A computer-aided diagnosis system for the classification of mammography mass lesions

Prerequisites

A Cuda enabled GPU (The implementation was tested on Ubuntu mate 16.04.1 LTS 64-bit and Nvidia GeForce GTX 980M graphic card)
Cuda 8.0 and CuDNN v5
Python 2.7-3.5
Opencv 3.1.0
Theano 0.8.2
Keras library from https://github.com/fchollet/keras

To run the framework

$ python BCSF.py

To compute the ROC curve and AUC for the independent dataset

$ python roc.py data

Name		Name	Last commit message	Last commit date
Latest commit History 10 Commits
B1(benign).png		B1(benign).png
B2(benign).png		B2(benign).png
B3(benign).png		B3(benign).png
BCSF.py		BCSF.py
LICENSE.md		LICENSE.md
M1(cancer).png		M1(cancer).png
M2(cancer).png		M2(cancer).png
M3(cancer).png		M3(cancer).png
README.md		README.md
blank.png		blank.png
dataset.py		dataset.py
model-labels.json		model-labels.json
model.h5		model.h5
model.json		model.json
model.png		model.png
net.py		net.py
roc.png		roc.png
roc.py		roc.py