Fungal_Keratitis_Classification

Installation

git clone https://github.com/IGLICT/Fungal_Keratitis_Classification.git
cd Fungal_Keratitis_Classification
pip install -r requirements.txt

Train the stage 1 network

python main_stage.py --stage='train' --config='./FungalKeratitis/stage1_swintransformer.yaml'

Test the stage 1 network

python main_stage.py --stage-'test' --config='./FungalKeratitis/stage1_swintransformer.yaml'

Name		Name	Last commit message	Last commit date
Latest commit History 4 Commits
Data		Data
FungalKeratitis		FungalKeratitis
datasets		datasets
models		models
.gitignore		.gitignore
README.md		README.md
main_stage.py		main_stage.py
requirements.txt		requirements.txt
utils.py		utils.py