search

PaddleCV-SIG / MedicalSeg

MedicalSeg is an easy-to-use 3D medical image segmentation toolkit that supports the whole segmentation process. Specially, We provide data preprocessing acceleration, high precision model on COVID-19 CT scans dataset and MRISpineSeg spine dataset, and a 3D visualization demo based on itkwidgets.
Apache License 2.0
66 stars 14 forks source link

[Feature] add nnunet val #84

Closed justld closed 2 years ago

justld commented 2 years ago

原代码train.py有个路径错误，已修改 移除了部分文件中无用的代码 添加验证代码，单fold验证命令：

python nnunet/single_fold_eval.py --config configs/msd/msd_lung_2d_fold_0.yml --model_path output/nnunet_2d/fold_0/iter_25000/model.pdparams --predict_next_stage --val_save_folder output/val/fold0 --precision fp16
shiyutang commented 2 years ago

这个PR是提交到develop分支的，是否先均提交到nnunet再提交到devvelop？这样能看到修改内容，同时测试部分是否修改好？

justld commented 2 years ago

经测试单折训练无误，单折验证无误，可以合入，接下来PR计划：添加五折验证-》ensemble-》五折预测