nnUNet 依然是最鲁棒的分割网络。今年初原作者更新了 nnUNet 的第二版即 nnUNetv2,题主近期试用的时候有一些踩坑的地方进行记录。
没有官方提供的预训练权重
官方文档在一个非常不起眼的角落提及了,v2版本尚无预训练权重可直接使用,有相关需求请使用 V1 版。
windows 下 nnUNet_raw 等参数设置
第一版 nnUNet 是不支持 windows 的,v2则添加了windows支持。在进行训练前需要设置 nnUNet_raw 等三个路径参数,如何设置可以参考其他教程。此处只提及如何设置三个路径。以 anaconda 为例,直接在对应虚拟环境中 set nnUNet_raw = 路径
即可,注意路径避免.
等特殊符号,此外不需要双引号,如 set nnUNet_raw = C:\\myPC\\Desktop
即可。