
nnunet
里清外明
familyfan,15879741247
展开
-
nnunetv2系列:使用onnx模型参数利用onnxruntime推理
作者分享的示例,这里在此基础上进行修改和增加了将预测结果转换到方便人查看的掩码图。原创 2024-09-25 18:39:20 · 618 阅读 · 0 评论 -
nnunetv2系列:多个2D图像推理
【代码】nnunetv2系列:多个2D图像推理。原创 2024-09-24 19:34:41 · 395 阅读 · 0 评论 -
nnunetv2系列:单个2D图像推理
对于单个图像,使用官方默认的推理方式会比较慢,而且在测试阶段推荐先用cpu而不是gpu进行推理,对于大批量的图像推荐使用官方默认的推理方式,且使用gpu。下面的代码同样根据官方给的示例进行调整得到推理单个2D图像的代码。相比默认的推理方式,这里支持图像名不以_0000.png结尾。为了方便人查看预测的效果,增加了恢复自定义的标签值模块。原创 2024-09-24 18:29:13 · 397 阅读 · 0 评论 -
nnunetv2系列: 导出模型参数文件和配置文件成zip压缩包
根据命令在文件中查找到信息,那么在文件中可以看到定义的相关的参数和默认值。我们也可以通过命令直接获取相关的参数信息。这里仅提供五折中的一折的模型参数文件和配置文件,如果需要全部的模型参数文件和配置文件,需要修改命令中的-f参数,默认是五折。原创 2024-09-13 07:00:00 · 322 阅读 · 0 评论 -
nnunetv2系列:解压zip文件到nnunet默认路径
这个命令的功能是将zip文件解压到路径下的指定位置,用于后续推理使用。在新的电脑上面使用同样需要安装nnUNet库以及配置环境。根据命令在文件中查找到信息,那么在文件中可以看到定义的相关的参数和默认值。我们也可以通过命令直接获取相关的参数信息。原创 2024-09-12 07:30:00 · 207 阅读 · 0 评论 -
nnunetv2系列:查看命令对应的函数
文件中定义了入口函数。原创 2024-09-12 07:00:00 · 310 阅读 · 0 评论 -
nnunetv2系列:2D实例分割数据集转换
这里主要参考官方源文件,注释了一些不必要的操作。数据集下载链接:nnU-Net只能用于使用无损(或无)压缩的文件格式!因为文件格式是为整个数据集定义的(而不是单独为图像和分割定义的,这可能是将来的任务),我们必须确保没有破坏分割映射的压缩工件。所以没有。jpg之类的!支持的2D数据集文件类型包括。原创 2024-09-11 19:25:16 · 1339 阅读 · 0 评论 -
nnunetv2系列:预测掩码可视化
实例分割的掩码输入的值,一般都得转成0, 1, 2这样的数值,预测出来后人无法直接可视化查看结果,这里给出代码示例,根据自定义的值进行转换,适合单通道灰度图和三通道彩色图。原创 2024-09-11 07:30:00 · 412 阅读 · 0 评论 -
nnunetv2系列:使用默认的预测类推理2D数据
这里参考源代码中给的示例进行调整和测试。原创 2024-09-10 20:50:49 · 1033 阅读 · 0 评论 -
nnunetv2系列:torch转onnx
作者提供的代码,这里做了一些简单的调整和优化,原代码存在覆盖文件问题。原创 2024-09-10 20:41:06 · 612 阅读 · 3 评论