Ir

.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

IR/IR.cpp

@@ -0,0 +1,34 @@
+#include "IR.h"
+
+IR::IR(std::string filename, Model model): _graph(){
+    _fileName = filename;
+    _model = model;
+    switch (_model)
+    {
+    case Model::ONNX:
+        _modelParser = new ONNX_ModelParser(_fileName);
+    break;
+
+    case Model::OPENCV:
+        _modelParser = new OPENCV_ModelParser(_fileName);
+    break;
+
+    // case Model::PYTORCH:
+    //     _modelParser = new OPENCV_ModelParser(_fileName);
+    //     break;
+    default:
+        break;
+    }
+
+    _graph = _modelParser->Parse();
+}
+
+const bool IR::operator==(const Graph& const) const {
+    return 0;
+}
+
+
+IR::~IR(){
+    delete _modelParser;
+    _modelParser = nullptr;
+}

IR/IR.h

@@ -0,0 +1,76 @@
+#include "graph.h"
+#include "map"
+
+class AnyLayer : protected Layer
+{
+private:
+    std::string _name;
+    std::map<std::string, std::string> _attribute;
+    std::vector<float> weights;
+    std::vector<float> bias;   
+
+public:
+    AnyLayer();
+    ~AnyLayer() = default; 
+
+    void addNeighbor(Layer* neighbor);
+    void removeNeighbor(Layer* neighbor);
+    std::list<Layer*> neighbors_; 
+};
+
+enum Model{ONNX, PYTORCH, OPENCV};
+
+class ModelParser {
+protected:
+    std::string _fileName;
+
+public:
+    explicit ModelParser(std::string filename) : _fileName(std::move(filename)) {}
+    virtual ~ModelParser() = default;
+    virtual Graph Parse() = 0;
+};
+
+class ONNX_ModelParser : public ModelParser {
+public:
+    explicit ONNX_ModelParser(std::string filename) : ModelParser(std::move(filename)) {}
+    Graph Parse() override;
+};
+
+class OPENCV_ModelParser : public ModelParser {
+public:
+    explicit OPENCV_ModelParser(std::string filename) : ModelParser(std::move(filename)) {}
+    Graph Parse() override; 
+};
+
+// class PYTORCH_ModelParser : ModelParser{
+// public:
+    // explicit OPENCV_ModelParser(std::string filename) : ModelParser(std::move(filename)) {}
+    // Graph Parse() override; 
+// };
+
+
+class IR
+{
+private:
+    std::string _fileName;
+    Graph _graph;
+    Model _model;
+    ModelParser* _modelParser;
+
+public:
+
+    IR(std::string filename, Model model);
+
+    const bool operator==(const Graph& graph) const;
+
+    // void ChangeModelType(Model model){
+    //     _model = model;
+    // }
+
+    // void ChangeFile(std::string filename){
+    //     _fileName = filename;
+    // }
+
+    ~IR();
+};
+

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