SegMitos_mitosis_detection
Weakly Supervised Mitosis Detection in Breast Histopathology Images using Concentric Loss By Chao Li, Xinggang Wang, Wenyu Liu, Longin Jan Latecki, Bo Wang and Junzhou Huang
Codes for our Medical Image Analysis paper "Weakly Supervised Mitosis Detection in Breast Histopathology Images using Concentric Loss".
Contents
Requirements: software
We use the deep learning framework Caffe to train the segmentation network. The code for data preparation is written by Matlab, while the codes for training the SegMitos network and detecting the mitosis is written in Python.
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.
Basic steps
-
Install the Caffe framework. Note that our model uses the old version of Caffe crop layer, which doesn't have "crop_param" parameter. Specifically, we use the Caffe of HED. Latest version of Caffe should also work, but the parameters of crop layer need to be added.
-
Prepare data and produce the concentric labels. Please see
data/README.md
for more details. -
Train the SegMitos model:
cd SegMitos_train/AMIDA13-fcn32s python solve.py
-
Deploy the trained model on the testing images of mitosis dataset.
cd SegMitos_detection/ python Demo.py