Add support for best checkpoints + update `download_models`

axondeepseg / nn-axondeepseg

1-class and 2-class segmentation of axon/myelin using nnunetv2

1 stars 0 forks source link

Add support for best checkpoints + update `download_models` #12

Closed hermancollin closed 6 months ago

hermancollin commented 6 months ago

The goal of this PR is to update the download_models script and fix the bug reported in #8 (inference script fails when final_checkpoints are not provided).

Solves #10 Solves #11