pycaffe无法识别prototxt文件中的有效参数

时间:2016-12-12 15:50:45

标签: python neural-network caffe conv-neural-network pycaffe

我正在试图了解卷积神经网络SegNet的作用,所以我创建了一个Python模块,希望能够打印出不同形状和属性的层。

var selectbox = document.getElementsByClassName('select');
for(var i = 0; i < selectbox.length; i++) {
    selectbox[i].onclick = function() {
        var elechild = this.childNodes;
        var x = 0;
        for(x; x < elechild.length; x++) {
            elechild[x].onclick = function() {
                console.log(this.getAttribute('data-select'));
            }
        }
    }
}

但它告诉我该文件不正确,尽管我可以使用SegNet提供的文件正确训练此网络(请参阅原型文件here)。这意味着什么,我该怎么做?

这是完整的错误

from __future__ import print_function
import numpy as np
import matplotlib.pyplot as plt
from PIL import Image
import caffe

caffe.set_device(0)
caffe.set_mode_gpu()
net = caffe.Net('/path/to/models/segnet_train.prototxt', caffe.TEST)

[print(k, v.data.shape) for k, v in net.blobs.items()]

1 个答案:

答案 0 :(得分:1)

正如戴尔在他的comment中所指出的,这确实是一个咖啡版本的问题。我确实使用了独立的caffe构建,而SegNet则带有自己的caffe。

我将python路径更改为此分发,一切正常。