Repository dedicated to training retinal vessel segmentation models
inside docker/ folder
- container configuration with port mappings and container naming can be found in container_source file
- ./build.sh
- container configuration with port mappings and container naming can be found in container_source file
- ./container_up.sh
- ./bash_exec.sh
- ./container_logs.sh
- ./jupyter_exec.sh
- ./container_rm.sh
execute everything in running docker container
- configure experiment in config.py
- ./train.sh 0 first argument has to be a number pointing to an existing gpu_id in PCI_BUS_ID order
Experiment configuration, logs, epoch metrics, tensorboard data and checkpoints will be stored in experiment folder in experiments/
- ./eval.sh <path to experiment configuration file .json> <device_id>
Evaluation result will be placed in selected model's experiment folder. Submission .zip file will be created too.