阅读nnUNet\nnunetv2\preprocessing\preprocessors\default_preprocessor.py文件
文件内有一个DefaultPreprocessor类和example_test_case_preprocessing函数(测试用的,跳过),阅读DefaultPreprocessor类。
在DefaultPreprocessor类内涉及的其他函数都在文章后半部分说明。
数据预处理是在生成dataset_fingerprint.json文件和生成nnUNetPlans.json文件后根据这两个文件内的信息操作的。
__init__函数
定义verbose,该变量用于确定是否打印更多信息的
run函数
参数
- dataset_name_or_id:数据集名称或id
-
configuration_name:配置的名称,例如2d
-
plans_identifier:默认nnUNetPlans
-
num_processes:进程数
过程
逻辑和生成dataset_fingerprint.json文件差不多。
配置一些文件名称、路径,加载(读取)nnUNetPlans.json文件,实例化PlansManager类,读取配置信息(get_configuration函数)等。
最后是多进程处理数据,逻辑和