Hello World (Witaj, świecie) w C++

Sprawdź, czy masz działającą wersję MediaPipe Framework. Zapoznaj się z instrukcjami instalacji.

Aby uruchomić przykład hello world:

$ git clone https://github.com/google/mediapipe.git
$ cd mediapipe

$ export GLOG_logtostderr=1
# Need bazel flag 'MEDIAPIPE_DISABLE_GPU=1' as desktop GPU is not supported currently.
$ bazel run --define MEDIAPIPE_DISABLE_GPU=1 \
    mediapipe/examples/desktop/hello_world:hello_world

# It should print 10 rows of Hello World!
# Hello World!
# Hello World!
# Hello World!
# Hello World!
# Hello World!
# Hello World!
# Hello World!
# Hello World!
# Hello World!
# Hello World!

W przykładzie hello world użyto prostego wykresu MediaPipe w funkcji PrintHelloWorld() zdefiniowanej w protokole CalculatorGraphConfig.

absl::Status PrintHelloWorld() {
  // Configures a simple graph, which concatenates 2 PassThroughCalculators.
  CalculatorGraphConfig config = ParseTextProtoOrDie<CalculatorGraphConfig>(R"(
    input_stream: "in"
    output_stream: "out"
    node {
      calculator: "PassThroughCalculator"
      input_stream: "in"
      output_stream: "out1"
    }
    node {
      calculator: "PassThroughCalculator"
      input_stream: "out1"
      output_stream: "out"
    }
  )");

Możesz zwizualizować ten wykres za pomocą narzędzia MediaPipe Visualizer, wklejając w wizualizatorze zawartość z CalculatorGraphConfig. Pomoc dotyczącą wizualizacji znajdziesz tutaj.

    input_stream: "in"
    output_stream: "out"
    node {
      calculator: "PassThroughCalculator"
      input_stream: "in"
      output_stream: "out1"
    }
    node {
      calculator: "PassThroughCalculator"
      input_stream: "out1"
      output_stream: "out"
    }

Ten wykres składa się z 1 grafu wejściowego strumienia wejściowego (in) i 1 strumienia wyjściowego wykresu (out) oraz 2 połączonych szeregowo strumieni PassThroughCalculator.

Przed uruchomieniem wykresu obiekt OutputStreamPoller jest połączony ze strumieniem danych wyjściowych, by później pobrać dane wyjściowe wykresu. Uruchomienie wykresu rozpoczyna się od funkcji StartRun.

CalculatorGraph graph;
MP_RETURN_IF_ERROR(graph.Initialize(config));
MP_ASSIGN_OR_RETURN(OutputStreamPoller poller,
                    graph.AddOutputStreamPoller("out"));
MP_RETURN_IF_ERROR(graph.StartRun({}));

Następnie przy użyciu funkcji MakePacket tworzymy 10 pakietów (każdy pakiet zawiera ciąg „Hello World!” z wartościami sygnatury czasowej z zakresu od 0, 1 do ... 9), dodaje każdy pakiet do wykresu przez strumień wejściowy in i zamyka strumień, aby zakończyć działanie wykresu.

for (int i = 0; i < 10; ++i) {
  MP_RETURN_IF_ERROR(graph.AddPacketToInputStream("in",
                     MakePacket<std::string>("Hello World!").At(Timestamp(i))));
}
MP_RETURN_IF_ERROR(graph.CloseInputStream("in"));

Dzięki obiektowi OutputStreamPoller przykład pobiera wszystkie 10 pakietów ze strumienia danych wyjściowych, pobiera ciąg znaków z każdego pakietu i drukuje go w logu wyjściowym.

mediapipe::Packet packet;
while (poller.Next(&packet)) {
  LOG(INFO) << packet.Get<string>();
}