Skip to content
Multitask computer aided diagnosis of breast masses
Python
Branch: master
Clone or download
Pull request Compare This branch is 2 commits ahead of Ivylab:master.
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
preprocess
.gitignore
CODE_OF_CONDUCT.md
CONTRIBUTING.md
LICENSE
README.md
main_ICADx.py
model_vgg.py

README.md

Multitask computer aided diagnosis of breast masses

In this page, a multitask computer aided diagnosis (CADx) framework is devised to investigate interpretability for classifying breast masses. We investigate interpretability in CADx with the proposed multitask CADx framework.

image_2018-11-28 23_07_21

Dependencies

  • MATLAB2014
  • tensorflow >= 0.10.0rc0
  • h5py 2.3.1
  • scipy 0.18.0
  • Pillow 3.1.0
  • opencv2

Pretrained Models

Thanks to Davi Frossard you can download a pretrained weight here https://www.cs.toronto.edu/~frossard/vgg16/vgg16_weights.npz.

How to use

First, go into the 'data' directory and download "DDSM dataset"

For data consistency, mammograms scanned by Howtek 960 were selected from the DDSM dataset. By using LJPEG2PNG tool "DDSMUtility", load breast image. Preprocess the DDSM dataset to create a h5 file. It resizes images to 64*64.

After that preprecess the pretrained model vgg16_weights.npz

cd preprocess
python prepare.py

Finally, run 'main_ICADx.py'.

cd ..
python main_ICADx.py

You can see the samples in 'img and test' directory.

Reference Code

https://github.com/trane293/DDSMUtility

You can’t perform that action at this time.