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:
@@ -59,7 +65,20 @@ jobs:
         submodules: recursive
     - name: Install dependencies
       run: |
-        brew install ninja scons
+        brew install ninja scons 
+
+
+
+
+
+
+
+
+
+
+
+        python -m pip install --upgrade pip
+        python -m pip install -r requirements.txt
     - name: ccache
       uses: hendrikmuhs/[email protected]
       with:
@@ -82,6 +101,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(frontend)
 add_subdirectory(example)

app/frontend/CMakeLists.txt

@@ -0,0 +1,49 @@
+cmake_minimum_required(VERSION 3.10)
+project(ModelParser)
+
+set(CMAKE_CXX_STANDARD 11)
+set(CMAKE_CXX_STANDARD_REQUIRED ON)
+
+find_package(Protobuf REQUIRED)
+
+set(ONNX_GENERATED_DIR "${CMAKE_SOURCE_DIR}/generated") 
+
+set(ONNX_PROTO_SRCS
+    "${ONNX_GENERATED_DIR}/onnx.pb.cc"
+)
+
+set(ONNX_GENERATED_HEADER 
+    "${ONNX_GENERATED_DIR}/onnx.pb.h"
+) 
+
+set(SRC_FILES
+    main.cpp
+    ${ONNX_PROTO_SRCS}
+)
+
+set_source_files_properties(${CMAKE_CURRENT_BINARY_DIR}/generated/onnx.pb.h PROPERTIES GENERATED TRUE)
+set_source_files_properties(${CMAKE_CURRENT_BINARY_DIR}/generated/onnx.pb.cc PROPERTIES GENERATED TRUE)
+set_source_files_properties(${CMAKE_CURRENT_BINARY_DIR}/generated/yolo11x.onnx PROPERTIES GENERATED TRUE)
+
+file(MAKE_DIRECTORY ${CMAKE_CURRENT_BINARY_DIR}/generated)
+
+set(YOLO_SRC "${CMAKE_SOURCE_DIR}/generated/yolo11x.onnx")
+set(YOLO_DST "${CMAKE_CURRENT_BINARY_DIR}/generated/yolo11x.onnx")
+
+if(EXISTS "${YOLO_SRC}")
+    configure_file(${YOLO_SRC} ${YOLO_DST} COPYONLY)
+else()
+    file(WRITE "${YOLO_DST}" "")
+endif()
+
+add_executable(ModelParser ${SRC_FILES}) 
+
+target_include_directories(ModelParser PRIVATE
+    ${ONNX_GENERATED_DIR} 
+    ${Protobuf_INCLUDE_DIRS}
+)
+
+target_link_libraries(ModelParser 
+    ${Protobuf_LIBRARIES}   
+    ${ONNX_GENERATED_HEADER}
+)

app/frontend/GetModel.sh

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

app/frontend/README.md

@@ -0,0 +1,19 @@
+## __How to run yolov11x model parser__
+
+1. Build the project.
+
+2. Get a network model using Ultralytics CLI
+
+-   Give "executable mode" to the script GetModel.sh 
+    ```
+    chmod +x GetModel.sh
+    ```
+-   Run script
+    ```
+    bash GetModel.sh
+    ```
+
+3. Run the project 
+    ```
+    ./ModelParser
+    ```

app/frontend/main.cpp

@@ -0,0 +1,35 @@
+#include <fstream>
+#include <iostream>
+#include <string>
+#include <vector>
+
+#include "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.32