Как получить OpenCV-подобный выход из сети Caffe? - PullRequest
Новая
       15

Как получить OpenCV-подобный выход из сети Caffe?

0 голосов
/ 13 июня 2019

Я использовал dnn модуль OpenCV для обнаружения OpenPose.Я загрузил предварительно обученную модель caffe (.caffemodel), получил вывод в формате Mat и затем получил ключевые точки из этого, как в этом примере: https://docs.opencv.org/3.4/d7/d4f/samples_2dnn_2openpose_8cpp-example.html

Теперь я пытаюсь использовать Caffe, но я не знаю, какполучить Mat из caffe-network для обнаружения ключевых точек.

Я пытаюсь это сделать, но это не работает. 

const Blob<float>* blob = net_->output_blobs()[0];

const int channels = blob->channels();
const int height = blob->height();
const int width = blob->width();
const cv::Size blobSize(width, height);

float* data = blob->mutable_cpu_data();

for (int i = 0; i < channels; ++i)
{
     cv::Mat channel(blobSize, CV_32FC1, data);
     //work with Mat
     data += blobSize.area();
}

Как я могу использовать Caffe Framework как модуль DNN из OpenCV?Или как я могу получить ключевые точки от этой модели?Спасибо.

...