daviddmc / NeSVoR

NeSVoR is a package for GPU-accelerated slice-to-volume reconstruction.
MIT License
72 stars 17 forks source link

Wrong checkpoint for DynUNet #7

Closed Edenzzzz closed 1 year ago

Edenzzzz commented 1 year ago

I ran nesvor segment-stack --input-stacks volumes/case2_day1_stack.nii.gz --output-stack-masks volumes/case2_day1_mask.nii.gz but the downloaded checkpoint (nesvor/preprocessing/masking/brain_segmentation.py) is completely wrong and all keys are unexpected.

image

daviddmc commented 1 year ago

Hi @Edenzzzz, did you use the docker or install NeSVoR from source? The segmentation model (i.e., MONAIfbs) uses monai v0.3.0. Could you check if your version of monai is too high?

Edenzzzz commented 1 year ago

Thanks for the speedy reply. I installed from source with MONAI 1.2. Switching to V0.3 works fine.