Пакетная обработка с использованием OpenCV "blobFromImages" Проблема - net .forward () не работает (UNet Архитектура) (C ++) - PullRequest
0 голосов
/ 19 июня 2020

Я хочу выполнить пакетную обработку с помощью OpenCV. Я написал код, чтобы сделать все и пройти все, кроме моей функции net .forward (). Также хорошо отметить, что это работает с «blobFromImage» нормально, однако терпит неудачу только тогда, когда я пытаюсь использовать пакетную обработку с «blobFromImages». Моя сеть - это UNet, выполняющая семанти c сегментацию. Для rea dNet я использую readNetFromTensorFlow. OpenCV версии 4.3.0.

std::vector<cd::Mat> Images
Images.push_back(image1)
Images.push_back(image2)

cv::Mat blob = cv::dnn::blobFromImages(Images, scale, cv::Size())
net.setInput(blob)
//Code Fails Here
cv::Mat OutputPreds = net.forward()

Я получаю ожидаемый размер капли [N, C, H, W]. Что-то еще мне здесь не хватает? Я также попытался установить OutputPreds как "std :: vector", но это тоже не удалось.

...