axondeepseg / model_seg_generalist

Axon and myelin segmentation model trained on all ADS datasets. Works across contrasts and resolutions.
1 stars 0 forks source link

Enhanced Inference and Evaluation Capabilities for nnU-Net Model #15

Closed ArthurBoschet closed 7 months ago

ArthurBoschet commented 8 months ago

PR: Enhanced Inference and Evaluation Capabilities for nnU-Net Models

This PR introduces enhancements to the inference and evaluation processes of nnU-Net models, specifically targeting the functionality to execute these processes on individual folds in addition to the existing ensembling approach. Below is a summary of the key changes incorporated into this update:

Changes

Goal

The primary goal of this PR is to enhance the flexibility and effectiveness of the inference and evaluation stages of nnU-Net models. By allowing individual folds to be evaluated in addition to the ensemble of folds, users can gain deeper insights into their models' performance and make more informed decisions regarding their deployment.