使用 nnUNetv2 的一些踩坑记录(或许还会有继续更新

nnUNet 依然是最鲁棒的分割网络。今年初原作者更新了 nnUNet 的第二版即 nnUNetv2,题主近期试用的时候有一些踩坑的地方进行记录。

没有官方提供的预训练权重

官方文档在一个非常不起眼的角落提及了,v2版本尚无预训练权重可直接使用,有相关需求请使用 V1 版。

windows 下 nnUNet_raw 等参数设置

第一版 nnUNet 是不支持 windows 的,v2则添加了windows支持。在进行训练前需要设置 nnUNet_raw 等三个路径参数,如何设置可以参考其他教程。此处只提及如何设置三个路径。以 anaconda 为例,直接在对应虚拟环境中 set nnUNet_raw = 路径 即可,注意路径避免. 等特殊符号,此外不需要双引号,如 set nnUNet_raw = C:\\myPC\\Desktop 即可。

未完待续