Размер ввода нейронной сети в opencv :: dnn

Я хотел бы определить (динамически) ожидаемый размер изображения как вход модели глубокой нейронной сети, которая загружается с dnn модуль opencv, Например, я загружаю caffe Модель у меня сначала есть:

cv::Mat img = imread("my_image.jpg");
cv::dnn::Net m_net;
m_net = cv::dnn::readNetFromCaffe("my_prototxt","my_model");

int img_size; // TODO how to determine it ?

cv::Mat blob =  cv::dnn::blobFromImage(img, 1.0, cv::Size(img_size, img_size));
m_net.setInput(blob);
cv::Mat res = m_net.forward("layer_name");

Моя проблема заключается в определении img_size fomr m_net Вот. Я думаю, это должно быть осуществимо с dnn::getLayersShapes но я не понимаю, что должно быть первым аргументом в моем случае.

Я привожу здесь пример модели кафе, но меня интересует возможность сделать то же самое с tensorflow, darknet или же torch модель также.

0

Решение

Задача ещё не решена.

Другие решения

Других решений пока нет …