首页
学习
活动
专区
圈层
工具
发布
首页
学习
活动
专区
圈层
工具
MCP广场
社区首页 >问答首页 >如何访问NiftyNet中预训练模型的中间激活图?

如何访问NiftyNet中预训练模型的中间激活图?
EN

Stack Overflow用户
提问于 2017-10-20 02:16:04
回答 1查看 625关注 0票数 1

我可以下载并成功测试NiftyNet包的brain parcellation demo。然而,这只给了我一个预先训练的网络的最终分割结果,而我也需要访问中间层的输出。

根据这个演示,下面这行代码下载了一个预先训练好的模型和一个测试MR卷:

代码语言:javascript
运行
复制
wget -c https://www.dropbox.com/s/rxhluo9sub7ewlp/parcellation_demo.tar.gz -P ${demopath}

其中${demopath}是演示文件夹的路径。解压下载的文件将创建一个.ckpt文件,该文件似乎包含预训练的tensorflow模型,但是我无法将其加载到tensorflow会话中。

有没有一种方法可以加载预先训练的模型并访问它的所有中间激活图?换句话说,我如何将预先训练好的模型从NiftyNet库加载到tensorflow会话中,这样我就可以浏览模型或探索任何给定输入图像的特定中间层?

最后,在NiftyNet的网站上提到,“文献中的一些模型已经(重新)在NiftyNet框架中实现了”。这些模型的预训练权重是否也可用?该演示使用了一个名为HighRes3DNet的预训练模型。如果其他模型的预训练权重也可用,那么下载这些权重或保存的tensorflow模型的链接是什么?

EN

回答 1

Stack Overflow用户

发布于 2017-10-20 15:25:42

首先回答你的“最后”问题,NiftyNet已经实现了一些网络架构(例如,VNet,UNet,DeepMedic,HighRes3DNet),你可以在自己的数据上进行训练。对于其中的一小部分,有针对某些应用的预先训练的权重(例如,使用HighRes3DNet的脑部分割和使用DenseVNet的腹部CT分割)。

这些预先训练的权重中的一些是从演示中链接的,比如您链接到的parcellation。我们开始将预先训练好的模型收集到model zoo中,但这仍然是一项正在进行的工作。

Eli Gibson NiftyNet开发人员

票数 2
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/46836389

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档