Skip to content

Source code for the paper "Automated Detection of Peripheral Artery Disease From High Resolution Color Fundus Photographs" by Mueller, S et al.

Notifications You must be signed in to change notification settings

AgenoDrei/pad_classification

Repository files navigation

Source code for the paper "Automated Detection of Peripheral Artery Disease From High Resolution Color Fundus Photographs"

by Mueller, S et al.

File structure

  • include/ - scirpts providing classes for the creation of models, datasets and general convenience functions
  • scripts/ - standalone scripts that are used to prepare the data or generate visualizations
  • run_k_fold.py - main script to run the training using cross-validation
  • search_hyperparameters.py - script to search for optimal hyperparameters across one fold in the dataset
  • multiple_instance_learning.py - script to run the training process for the MIL model on a training and validation set
  • prepare_data.py - script to convert the raw dataset into folds and generate label files for the training

All training scripts are configured using the *.toml files.

Example

Example for executing the k-fold training

python run_k_fold.py --data <path_to_data> --epochs 20 -k 7 -s "MIL" --model <path_to_pretrained_model>

About

Source code for the paper "Automated Detection of Peripheral Artery Disease From High Resolution Color Fundus Photographs" by Mueller, S et al.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published