bowang-lab / U-Mamba

U-Mamba: Enhancing Long-range Dependency for Biomedical Image Segmentation

Home Page:https://arxiv.org/abs/2401.04722

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

How to do model inference and obtain the results in the paper

TicTacToePP opened this issue · comments

As the output folder only contains fold_all, I add a -f all after the provided inference command.

The command used for testing:
nnUNetv2_predict -i ./data/nnUNet_results/Dataset704_Endovis17 -o ./test_endo -d 704 -c 2d -tr nnUNetTrainerUMambaBot --disable_tta -f all

The results:
image

I managed to solve by myself to successfully get the results.

However, I am still confused by the standard deviation in the table. How to obtain them and why they have so large values?

image

commented

Hi @TicTacToePP ,

Thanks for your interest.

Did you try the evaluation code?

https://github.com/bowang-lab/U-Mamba/tree/main/evaluation

Hi @JunMa11,

Thanks for your response. Yes, I obtained the results with the evaluation code.
But I am wondering how to get the std values (after ±) in the table.

commented

Hi @TicTacToePP ,
The evaluation scripts generate CSV files containing the computed metric for each test case. The mean and standard deviation reported in our tables are calculated across all these test cases from the generated CSV files. For abdomen datasets with metric values of multiple organ classes reported in the CSV files, you can first calculate the average metric across classes for each case, and then calculate the standard deviation and mean across all test cases to obtain the reported means and standard deviations in the table.

由于输出文件夹仅包含fold_all,因此我在提供的推理命令后添加了 -f all。

用于测试的命令: nnUNetv2_predict -i ./data/nnUNet_results/Dataset704_Endovis17 -o ./test_endo -d 704 -c 2d -tr nnUNetTrainerUMambaBot --disable_tta -f all

结果: image

Excuse me, how did you solve this problem