如何在单通道图像上应用预先训练的3通道图像模型?

时间:2017-04-02 05:33:34

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

我尝试使用已经训练过的三通道彩色图像的预训练模型,然而,由于形状差异,我收到了错误。有人能让我知道我该如何解决这个问题? 一位用户建议使用<uses-permission android:name="android.permission.INTERNET" /> <uses-permission android:name="android.permission.ACCESS_NETWORK_STATE" /> <uses-permission android:name="android.permission.WRITE_EXTERNAL_STORAGE"/> 图层,但我找不到任何相关文档/帮助来使用此图层或任何其他解决方案。我非常感谢你的帮助。

1 个答案:

答案 0 :(得分:2)

caffe.proto关于图块层的信息不多。如果您查看代码,它只会为每个外部维度复制数据tiles次。对于你的情况应该是:

layer{
    name: "tile"
    type: "Tile"
    bottom: "bottom-blob"
    top: "top-blob"
    tiling_param {
        // axis is 1 by default
        tiles: 3
    }
}