8000 GitHub - andy-yu-ua/onnx_extractor_netron: Visualizer for neural network, deep learning and machine learning models
[go: up one dir, main page]
More Web Proxy on the site http://driver.im/
Skip to content

andy-yu-ua/onnx_extractor_netron

 
 

Repository files navigation

Sub_Netron

Sub_Netron is an extension of the Netron model viewer that lets you select nodes—either via double-click or by dragging a selection box (with the Shift key held)—and extract a subgraph from an ONNX model.

Requirements

Make sure you have Python installed. Then, install the required dependencies by running:

pip install -r requirements.txt

Running the Program

To launch the full Sub_Netron workflow (Netron interface + backend processing), run:

python run_subnetron.py <path_to_model.onnx>

This will:

  • Launch the Netron interface in your browser for model inspection.

  • Start the backend validation and extraction script.

Usage

  • Shift + Drag to draw a selection box and select nodes in the graph.

  • Click the Reset button to unselect all currently highlighted nodes.

  • After pressing Validate&Extract, the tool will send the selected node IDs and model path to the backend for validation and extraction.

Below is the Netron's README

Netron is a viewer for neural network, deep learning and machine learning models.

Netron supports ONNX, TensorFlow Lite, Core ML, Keras, Caffe, Darknet, PyTorch, TensorFlow.js, Safetensors and NumPy.

Netron has experimental support for TorchScript, TensorFlow, MXNet, OpenVINO, RKNN, ML.NET, ncnn, MNN, PaddlePaddle, GGUF and scikit-learn.

Install

macOS: Download the .dmg file or run brew install --cask netron

Linux: Download the .AppImage file or run snap install netron

Windows: Download the .exe installer or run winget install -s winget netron

Browser: Start the browser version.

Python: Run pip install netron and netron [FILE] or netron.start('[FILE]').

Models

Sample model files to download or open using the browser version:

About

Visualizer for neural network, deep learning and machine learning models

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • JavaScript 95.9%
  • Python 2.6%
  • Other 1.5%
0