Я хотел бы определить (динамически) ожидаемый размер изображения как вход модели глубокой нейронной сети, которая загружается с 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
модель также.
Задача ещё не решена.
Других решений пока нет …