XRad-Ulm / Proto-Caps

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Proto-Caps

Implementation code of paper "Interpretable Medical Image Classification using Prototype Learning and Privileged Information", MICCAI 2023

Quick start:

Set up pylidc, install requirements and run main.py with argument --train=True After training is finished, run main.py with arguments --test=True --model_path=<model_path> --epoch=<epoch_number_of_model>

Find more parser arguments in main.py to adapt model architecture and algorithm.

If you have any questions, please contact luisa.gallee@uni-ulm.de.

About

License:MIT License


Languages

Language:Python 100.0%