Computer aided detection using Faster-RCNN
Switch branches/tags
Nothing to show
Clone or download
Latest commit 6c2deb5 Mar 21, 2018
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
data Update README.md Nov 7, 2017
docs smaller plugin image Mar 14, 2018
results Update README.md Nov 7, 2017
README.md Update README.md Mar 21, 2018
_config.yml Set theme jekyll-theme-cayman Mar 14, 2018
demo.ipynb demo notebook Oct 26, 2017
vgg16_frcnn_cad_test.prototxt uploaded prototxt Oct 26, 2017

README.md

Computer aided detection with Faster-RCNN

A VGG16 based Faster-RCNN detector trained to detect breast cancer lesions on mammogram images.

This is the basis of the solution which reached 2nd place in The Digital Mammography DREAM Challenge.

For more details and results please read the article: https://www.nature.com/articles/s41598-018-22437-z


Model weights

Model weights can de downloaded from here:


OsiriX/Horos plugin

plugin

I wrote a plugin for the popular medical image viewer: OsiriX/Horos. This is a simplified version of the model. It is based on CoreML. In order to run it you will need:

  • A mac
  • High Sierra (10.13) operating system ( becase of CoreML )
  • Horos (free) or OsiriX medical image viewer.

Installation: Download the plugin, unzip it, click on the installer, and approve it. Restart OsiriX/Horos. The plugin will appear in the menu of OsiriX/Horos: Plugins/Image Filters/CnnCAD. You can also add it to the toolbar of the 2D viewer.

Usage: Open the images in the 2D viewer and click on the menu item: Plugins/Image Filters/CnnCAD. A pop-up will appear to remind you that the tool is not for medical use, and to warn you that detection may take minutes depending on your computer. Just click OK, wait until the beachball stops spinning and the ROI markers appear on the image (In case the model found something, I recommend to run the first test with an obvious cancer).

Note: The plugin is only tested with Horos. OsiriX MD may not work correctly at this time.

Note: The plugin does not work correctly with OsiriX Lite! OsiriX Lite exposes only a downsampled image of the mammogram, and as a result the marks are radically shifted towards the upper left corner. I do not wish to solve this issue with OsiriX Lite, firstly because mammograms should be viewed at full resolution, and also because the program is unusable anyway due to extremely annoying popups.

The OsiriX/Horos plugin can be downloaded from here:


If you want to use the model or the plugin in your research please cite:

Ribli, Dezső, et al. "Detecting and classifying lesions in mammograms with Deep Learning." Scientific Reports 8.1 (2018): 4165.