Home

Awesome

OnnxGraphQt

ONNX model visualizer. You can edit model structure with GUI!

https://github.com/fateshelled/OnnxGraphQt/blob/main/LICENSE https://github.com/fateshelled/OnnxGraphQt/stargazers

<p align="center"> <img src="https://user-images.githubusercontent.com/53618876/173075283-3344ca39-adcc-4e73-a5ea-31148fa641bf.png" /> </p>

Requirements

Install

sudo apt install python3-pyside2*

git clone https://github.com/fateshelled/OnnxGraphQt
cd OnnxGraphQt
python3 -m pip install -U nvidia-pyindex
python3 -m pip install -U Qt.py
# If you want to use InferenceTest, install onnxruntime or onnxruntime-gpu
# python3 -m pip install -U onnxruntime
# python3 -m pip install -U onnxruntime-gpu
python3 -m pip install -U -r requirements.txt

# Install OnnxGraphQt
python3 -m pip install .

Run with Docker

git clone https://github.com/fateshelled/OnnxGraphQt
cd OnnxGraphQt
# build docker image
./docker/build.bash
# run
./docker/run.bash

Usage

# Open empty graph
onnxgraphqt

# Open with onnx model
onnxgraphqt onnxgraphqt/data/mobilenetv2-7.onnx

mobilenetv7-7.onnx

Open Onnx Model

Open file dialog from menubar(File - Open) or drag and drop from file manager to main window.

Sample model is available at ONNXGraphQt/onnxgraphqt/data/mobilenetv2-7.onnx

file open

Export

Export to ONNX file or Json file.

Node detail

Double click on Node for more information.

node information

Node Search

Node search window can be open from menubar(View - Search). You can search node by name, type, input or output name.

serach

simple-onnx-processing-tools

Please refer to each tool's Github repository for detailed usage.

ToDo

References