Skip to content
master
Switch branches/tags
Code

Latest commit

 

Git stats

Files

Permalink
Failed to load latest commit information.
Type
Name
Latest commit message
Commit time
 
 
 
 
 
 
 
 
 
 

DeepMitosis: Mitosis detection via deep detection, verification and segmentation networks

By Chao Li, Xinggang Wang, Wenyu Liu and Longin Jan Latecki

Codes for our MIA (Medical Image Analysis) paper "DeepMitosis: Mitosis detection via deep detection, verification and segmentation networks". Please see the paper for more details.

System Overview

System overview

Results

Method 2012 MITOSIS test 2014 MITOSIS val set 2014 MITOSIS test set
DeepDet 0.832 0.489 -
DeepDet+Ver 0.817 - -
DeepDet+Seg+Ver - 0.572 0.437

Citing DeepMitosis

If you find DeepMitosis useful in your research, please consider citing:

@article{li2018deepmitosis,
    title = {DeepMitosis: Mitosis detection via deep detection, verification and segmentation networks},
    author={Li, Chao and Wang, Xinggang and Liu, Wenyu and Latecki, Longin Jan},
    journal={Medical image analysis},
    volume={45},
    pages={121--133},
    year={2018},
    publisher={Elsevier}
}

Contents

  1. Requirements: software
  2. Requirements: hardware
  3. Basic installation
  4. Data preparation

Requirements: software

All the three deep models use Caffe framework to train.

Our detection model is based on Faster R-CNN model. We use the py-faster-rcnn. You need to firstly install the py-faster-rcnn, more details see https://github.com/rbgirshick/py-faster-rcnn.

Our segmentation model is based on a FCN model derived from VGG-16.

The deep verification model is based on ResNet-50 architecture.

Requirements: hardware

We use a TITAN X GPU with ~12GB memory in our experiments. However, a good GPU with at least 8G of memory suffices.

Installation

1.Install the Caffe framework.

2.Install the py-faster-rcnn and train the DeepDet model on 2012 MITOSIS dataset.

3.Install the FCN (fully convolutional networks) to train the DeepSeg model on 2012 MITOSIS dataset, and then deploy DeepSeg on 2014 MITOSIS dataset.

4.Train a DeepVer model using the detection results produced by DeepDet on 2014 MITOSIS dataset.

Data preparation

Download the 2012 MITOSIS dataset and 2014 MITOSIS dataset.

For DeepDet model, the data are arranged as the VOC data in py-faster-rcnn. We transfer the mitosis' annotation to bounding box format.

For DeepSeg model, we convert the annotations to mask images.

We perform data augmentations and image crop to produce more training samples. The data augmentation includes image rotation, mirror. Please see our paper for more details. Noted that in detection model training, we remove the image patches that contain small cross-boundary mitotic cells from the training data.

About

Codes for our MIA paper "DeepMitosis: Mitosis detection via deep detection, verification and segmentation networks".

Resources

License

Releases

No releases published

Packages

No packages published