按照nnUNet官方下载配置好
准备好KiTS19数据,任务id设为40
1. 运行nnunet/dataset_conversion下的Task040_KiTS.py, 需要对Json的写入文件名进行修改,加上_0000。
其中base为存放KiTS19数据的本地路径,out为nnUNet_raw/nnUNet_raw_data/Task40_KiTS
生成如下文件夹imageTr, imageTs, labelTr, 以及dataset.json
2. 运行nnUNet_convert_decathlon_task -i /xxx/Task040_KiTS (上面新生成的路径,任务id会变成三位数)
3. 运行nnUNet_plan_and_preprocess -t 40 (40为任务id)
会生成一些用于训练的pkl配置文件,以及对数据进行预处理。 batch_size默认大小为2,可以在pkl文件中修改batch_size
4. 开始train