当前位置:  开发笔记 > 编程语言 > 正文

Caffe中的多个预训练网络

如何解决《Caffe中的多个预训练网络》经验,为你挑选了1个好方法。

是否有一种简单的方法(例如,不修改caffe代码)将多个预训练网络的重量加载到一个网络中?该网络包含一些与预训练网络具有相同尺寸和名称的图层.

我试图用NVidia DIGITS和Caffe实现这一目标.

编辑:我认为不可能直接从DIGITS这样做,正如答案所证实的那样.任何人都可以建议一种简单的方法来修改DIGITS代码,以便能够选择多个预训练网络吗?我检查了一下代码,并认为训练脚本是一个很好的起点,但我对Caffe没有深入的了解,所以我不确定实现这个目标的最佳/最快方法是什么.



1> Igor Ševo..:

正如Shai所说,没有办法做到这一点,所以我决定克隆官方存储库并进行适当的更改.我更改了代码,以便使用冒号作为分隔符加载多个预训练网络.

我在官方存储库上创建了一个pull请求,然后我的更改与DIGITS的主分支合并,这意味着现在可以在DIGITS中使用此功能.

推荐阅读
农大军乐团_697
这个屌丝很懒,什么也没留下!
DevBox开发工具箱 | 专业的在线开发工具网站    京公网安备 11010802040832号  |  京ICP备19059560号-6
Copyright © 1998 - 2020 DevBox.CN. All Rights Reserved devBox.cn 开发工具箱 版权所有