Проблема в получении выходных данных слоя FC в caffe (cpp) - PullRequest
0 голосов
/ 07 апреля 2019

Я пытаюсь найти вывод полностью связанного слоя.Я изменил ответ, данный в Как получить функции из нескольких слоев, используя c ++ в caffe немного и использовал в моем коде.Проблема в том, что для каждого тестового изображения я получаю один и тот же вывод слоя.Где я сделал не так?

    const boost::shared_ptr<Blob<float> > feature_blob = classifier.net_->blob_by_name("ip1");

    int count = feature_blob->channels() * feature_blob->height() * feature_blob->width();
    float* feature_array = new float[count]; 
    const float* feature_blob_data = feature_blob->cpu_data();

    memcpy(feature_array, feature_blob_data, count * sizeof(float));
...