Onnx model parser

.github/workflows/main.yml

@@ -16,6 +16,9 @@ jobs:
     - name: Install dependencies
       run: |
         sudo apt-get install -y cmake ninja-build ccache scons
+        sudo apt-get install python3-dev libprotobuf-dev protobuf-compiler
+        python -m pip install --upgrade pip
+        python -m pip install -r requirements.txt
     - name: ccache
       uses: hendrikmuhs/[email protected]
       with:
@@ -38,6 +41,9 @@ jobs:
     - name: Install dependencies
       run: |
         sudo apt-get install -y cmake ninja-build ccache scons
+        sudo apt-get install python3-dev libprotobuf-dev protobuf-compiler
+        python -m pip install --upgrade pip
+        python -m pip install -r requirements.txt
     - name: ccache
       uses: hendrikmuhs/[email protected]
       with:
@@ -82,6 +88,9 @@ jobs:
     - name: Install dependencies
       run: |
         sudo apt-get install -y cmake ninja-build ccache gcovr lcov scons
+        sudo apt-get install python3-dev libprotobuf-dev protobuf-compiler
+        python -m pip install --upgrade pip
+        python -m pip install -r requirements.txt
     - uses: actions/checkout@v4
       with:
         submodules: true

app/CMakeLists.txt

@@ -1 +1,2 @@
 add_subdirectory(example)
+add_subdirectory(frontend)

app/frontend/CMakeLists.txt

@@ -0,0 +1,51 @@
+cmake_minimum_required(VERSION 3.10)
+project(ModelParser)
+
+find_package(Protobuf REQUIRED)
+set(ONNX_GENERATED_DIR "${CMAKE_SOURCE_DIR}/app/frontend/build/generated")
+
+
+set(SETUP_SCRIPT "${CMAKE_SOURCE_DIR}/app/frontend/GetModel.sh")
+
+set(GENERATED_FILES
+"${ONNX_GENERATED_DIR}/onnx.pb.cc"
+"${ONNX_GENERATED_DIR}/onnx.pb.h"
+"${ONNX_GENERATED_DIR}/yolo11x.onnx"
+)
+
+
+add_custom_command(
+    OUTPUT ${GENERATED_FILES}
+    COMMAND ${CMAKE_COMMAND} -E make_directory "${ONNX_GENERATED_DIR}"
+    COMMAND ${CMAKE_COMMAND} -E chdir ${ONNX_GENERATED_DIR} bash ${SETUP_SCRIPT} "${ONNX_GENERATED_DIR}"
+    WORKING_DIRECTORY "${ONNX_GENERATED_DIR}"
+    DEPENDS ${SETUP_SCRIPT}
+    COMMENT "Create directory, generate ONNX-files and get YOLO model"
+    )
+
+add_custom_target(generate_output ALL
+DEPENDS ${GENERATED_FILES}
+)
+
+set(ONNX_GENERATED_DIR "${CMAKE_SOURCE_DIR}/app/frontend/build/generated")
+
+set_source_files_properties(
+  "${ONNX_GENERATED_DIR}/onnx.pb.cc"
+  "${ONNX_GENERATED_DIR}/onnx.pb.h"
+  "${ONNX_GENERATED_DIR}/yolo11x.onnx"
+  PROPERTIES GENERATED TRUE
+)
+
+add_executable(ModelParser
+  main.cpp
+  "${ONNX_GENERATED_DIR}/onnx.pb.cc"
+)
+add_dependencies(ModelParser generate_output)
+target_include_directories(ModelParser PRIVATE
+  "${ONNX_GENERATED_DIR}"
+  ${Protobuf_INCLUDE_DIRS}
+)
+
+target_link_libraries(ModelParser
+  ${Protobuf_LIBRARIES}
+)

app/frontend/GetModel.sh

@@ -0,0 +1,4 @@
+#!/bin/bash
+
+protoc --proto_path=3rdparty/onnx/onnx --cpp_out=$1 onnx.proto
+yolo export model=$1/yolo11x.pt format=onnx

app/frontend/main.cpp

@@ -0,0 +1,35 @@
+#include <fstream>
+#include <iostream>
+#include <string>
+#include <vector>
+
+#include "./generated/onnx.pb.h"
+
+int main() {
+  std::ifstream model_file("generated/yolo11x.onnx", std::ios::binary);
+
+  if (!model_file.is_open()) {
+    std::cerr << "Failed to open model" << std::endl;
+    return 1;
+  }
+
+  onnx::ModelProto model;
+  if (!model.ParseFromIstream(&model_file)) {
+    std::cerr << "Model parsing error" << std::endl;
+    return 1;
+  }
+  model_file.close();
+
+  std::vector<std::string> layer;
+
+  for (int i = 0; i < model.graph().node_size(); ++i) {
+    const onnx::NodeProto& node = model.graph().node(i);
+    layer.emplace_back(node.op_type());
+  }
+
+  for (auto it : layer) {
+    std::cout << it << std::endl;
+  }
+
+  return 0;
+}

requirements.txt

@@ -0,0 +1 @@
+ultralytics >= 8.3.107