Два входных потока на узел в медиапайпе - PullRequest
Новая
       0

Два входных потока на узел в медиапайпе

0 голосов
/ 15 января 2020

Я пытаюсь изменить пример ObjectDetection Google Mediapipe, добавив второй поток ввода в узел DetectionLabelIdToText. Я хочу отправить в калькулятор дополнительные данные (например, имена меток вместо загрузки из файла).

Я изменяю object_detection_mobile_gpu.pbtxt 

input_stream: "second_input"
...
node {
  calculator: "DetectionLabelIdToTextCalculator"
  input_stream: "filtered_detections"
  input_stream: "SECOND:second_input"
  output_stream: "output_detections"
  node_options: {
    [type.googleapis.com/mediapipe.DetectionLabelIdToTextCalculatorOptions] {
      label_map_path: "mediapipe/models/ssdlite_object_detection_labelmap.txt"
    }
  }
}
...

и добавляю 

cc->Inputs().Get("SECOND", 0).Set<int>();

в creation_label_id_to_text_calculator. cc.

Я не ожидаю никаких изменений и приложение будет работать без изменений, но после добавления второго входного потока в метод калькулятора узлов Process вообще не вызывается.

Что я делаю не так? Подобный случай объясняется в документации как пример калькулятора.

Ссылки: Пример кода Объясненный код калькулятора

...