netron-vscode-extension: A vscode extension for netron, support *.pdmodel, *.nb, *.onnx, *.pb, *.h5, *.tflite, *.pth, *.pt, *.mnn, *.tnnproto, *.param, etc. This code is based on NetronInVSCode but adds more model format support (such as *.pdmodel, *.nb, *.mnn, *.param, *.tnnproto, etc.) and some new Features. NOTE: This extension is mainly for personal use, so the development plans are based on individual needs and may not have intensive plans in the future.

Features

• add *.pdmodel, *.nb, *.mnn, *.param, *.tnnproto format support
• remove manually pip install requirements
• [WIP] onnx-simplifier support
• [WIP] paddle2onnx converter support
• [WIP] x2paddle converter support
• [WIP] MNN converter support
• [WIP] Optimize some UI components

Install

• Install netron-vscode-extension from vscode marketplace.

• Install Python 3.x on your computer. The step about netron installation follows is optional. This extension will call netron from python via node-python-bride to form an url which contains the visualized contents, then, pass it to this extension. netron-vscode-extension will check whether the netron is avaliable on your python and auto install it if necessary.

   python -m pip install netron  # optional

Usage

• Click on a saved model to see the following, such as picodet, yolov8n, etc.

Build

You can build the extension from sources:

• Open this repo in VS Code
• npm install
• F5 to start debugging

LICENSE

The codes is released under the MIT License.

Reference

• netron
• node-python-bride
• vscode-extension-samples
• NetronInVSCode