/cascade_transfer_learning_medical

github repository for paper "Deep Cascade Learning for Optimal Medical Image Classification via Transfer". MLHC2022

Primary LanguageJupyter Notebook

Cascade Transfer Learning

Github repository for paper "Deep Cascade Learning for Optimal Medical Image Feature Representation".

Installation

Please check out requirements.txt and install package accordingly.

Reproducing Figure 1(b)

check TCL_IDC.ipynb reproduce the result for Figure 1(b) in the paper. If want to train the network, please follow the step below:

Steps to train the network

First, download dataset in Here.

Second, Download trained models in Here and put them into SourceNetwork folder.

Third, run:

python TCL_IDC.py --root_dir='./Breast Histopathology Images' --network_address=./model/sourcemodel/SourceNetwork

License

MIT