nnunet
文章平均质量分 89
SetMaker
github:andre20000131
mail:lfchen@yeah.net
NUPT
展开
-
nnUNet推理与预测(手把手一步一步实现。接上文用自己的数据训练)
如上图所示,当我们训练了大概几轮之后会在nnUNet_trained_models文件下生成nnUNet以及2d的模型文件,我们可以看到fold_4里面会有model_best.model,这个模型文件就是我们训练好的模型了,我们也可以通过同级目录下的progress.png,来查看loss函数的变化。接上文有几点要注意的情况,首先nnunet并不需要手动调学习率和训练轮数(这里nnunet默认是要训练1000轮),学习率的调整是用的损失函数的加权平均值使用的损失函数来对其进行更正的。还有就是nnun...原创 2022-07-06 10:52:27 · 6000 阅读 · 12 评论 -
nnUNet使用教程(使用自己的数据训练自己的模型)
1.制作自己的数据集(使用自己的数据训练自己的模型)本文主要讲述了自己训练模型的复现,更多的关注在nnunet的具体使用,跟着文章一步一步的操作得到可以实际应用的模型。原创 2021-11-25 15:03:02 · 22989 阅读 · 24 评论