/BreastCancerCNN

https://teacher.bupt.edu.cn/zhuchuang/en/index.htm

Primary LanguageJupyter Notebook

Breast cancer histopathology image classification through assembling multiple compact CNNs Tweet

This is a toolbox that implements the testing of the approach described in our papers: Breast cancer histopathology image classification through assembling multiple compact CNNs.

Our experiment was carried out with caffe.

Overview

Network profile

deploy_SE_bn4.prototxt is the network profile of hybrid model in our paper "Breast Cancer Histopathology Image Classification through Assembling Multiple Compact CNNs"

deploy_64.prototxt and deploy_32.prototxt is the network profile of the method in "Breast Cancer Histopathological Image Classification using Convolutional Neural Networks"

Dataset (Available)

Filefold global and patch contains the image filelist. Download the BACH dataset first and modify the path in the txt file with your own path.

BreaKHis and BACH Challenge Dataset are utilized in our test.

You can also download the dataset and data splitting files in our paper for research purposes from Dataset-Downloading-Address.

Note: please follow the data usage regulations.

Pretrained models

You can get the pretrained models here. modelmodel-patch.

vote.ipynb

In this file you can extract the result of the global and patch images. After that, you can compute various indicators in the following vote.ipynb content.

addition: work11

We selected a small portion of the pictures on the BreakHis-400X dataset. We used them to reproduce the article "Breast Cancer Histopathology Image Classification through Assembling Multiple Compact CNNs". However, in this process, we faced some problems, the network can not fit the training set very well, so our accuracy of the dataset is very low, and can not achieve the effect in the article. Of course, this may also be caused by the data set we selected which is too small. We put the code here and everyone can check it out.

Citation

Please kindly cite our paper in your publications if it helps your research.

BMC Journal paper (testing on BreaKHis and BACH):

@article{zhu2019breast,
  title={Breast cancer histopathology image classification through assembling multiple compact CNNs},
  author={Zhu, Chuang and Song, Fangzhou and Wang, Ying and Dong, Huihui and Guo, Yao and Liu, Jun},
  journal={BMC Medical Informatics and Decision Making},
  volume={19},
  number={1},
  pages={198},
  year={2019},
  publisher={Springer}
}

Our ICIAR paper in BACH Challenge (testing on BACH Dataset):

@inproceedings{guo2018breast,
  title={Breast cancer histology image classification based on deep neural networks},
  author={Guo, Yao and Dong, Huihui and Song, Fangzhou and Zhu, Chuang and Liu, Jun},
  booktitle={International Conference Image Analysis and Recognition},
  pages={827--836},
  year={2018},
  organization={Springer}
}

Authors

Chuang Zhu, Huihui Dong, et.al.

If you have any questions, you can contact us directly.