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
68 stars 14 forks source link

[Feature] Add verify dataset code #45

Closed justld closed 2 years ago

justld commented 2 years ago

添加裁剪后数据检查的代码。 如果检查不通过，则assert异常。无返回值 sample：

from preprocess_utils import verify_dataset_integrity
verify_dataset_integrity(converted_data_dir)  # 传入nnUnet_convert_decathlon_task转换后的文件夹路径