/onnx-explorer

onnx explorer more features.

Primary LanguagePythonApache License 2.0Apache-2.0

onnx-explorer-logo

Design

onnx-explorer-design

Features

  • support export onnx model infos and summary[txt,json,csv]
  • support estimate model memory
  • support onnx model infer

Example

onnx model infos

onnx_explorer_3 onnx_explorer_4 image
Yolov5x6.txt Yolov5x6.json Yolov5x6.csv

onnx model summary

estimate model memory

onnx-explorer_2

onnx model infer

onnx-infer

Usage

Install

conda create -n onnx-explorer python=3.8
conda activate onnx-explorer
python3 -m pip install -r requirements.txt

onnx-explorer building...

Easy start run

sh run.sh
sh run_estimate.sh
sh infer.sh