Skip to content

Transfer Learning

haberlmatt edited this page Apr 14, 2020 · 1 revision

How to retrain a pre-trained network (aka. transfer learning)

The following instructions use the mito demo sample stored within this repository and a trained mitochondria model from the CDeep3M model zoo.

Quick run

cd ~

PreprocessTrainingData.py ~/cdeep3m/mito_testsample/training/images/ ~/cdeep3m/mito_testsample/training/labels/ ~/mito_AugTrData

runtraining.sh --additerations 500 --retrain $MODEL_FOLDER ~/mito_AugTrData ~/mito_retrainout`