Codes for our paper "Weakly Supervised Mitosis Detection in Breast Histopathology Images using Concentric Loss".
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 by Python.
We use a TITAN X GPU with ~12GB memory in our experiments. However, a good GPU with at least 8G of memory suffices.
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