deep-learning - Caffe 中的多个预训练网络

标签 deep-learning caffe conv-neural-network pycaffe nvidia-digits

有没有一种简单的方法(例如,不修改 代码)将多个预训练网络中的 wights 加载到一个网络中?该网络包含一些与预训练网络具有相同维度和名称的层。

我正在尝试使用 NVidia DIGITS 和 Caffe 来实现这一点。

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

最佳答案

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

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

关于deep-learning - Caffe 中的多个预训练网络,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/34466901/

相关文章:

python - Caffe:绘制 CNN 网络

python - 如何使用pycaffe重构caffe net

python-3.x - 使用 Keras 进行多类图像分类的多重预测

image - jpeg 压缩会影响使用卷积神经网络的训练和分类吗

python - RNN 参数没有更新?

python - tensorflow Keras : Can `Conv2d` layers now accept multispectral images meaning bands greater than 3

python - 如何提取/剪切模型分类的部分图像?

deep-learning - 深度学习的白化数据

classification - 构建平均图像文件时出错(Caffe)

python - 小图像分类任务的 Conv2d 层和滤波器的数量