axondeepseg / nn-axondeepseg

1-class and 2-class segmentation of axon/myelin using nnunetv2
1 stars 0 forks source link

Let user choose the best checkpoints #11

Closed hermancollin closed 6 months ago

hermancollin commented 6 months ago

This issue was found in #8. The inference script was originally (and naively) designed to use the final checkpoints, which are selected by nnunet by default. Most often, we will want the best checkpoints instead, but the user cannot specify this.