aidge-model-explorer 0.0.1

Aidge module for model explorer: https://github.com/google-ai-edge/model-explorer.

Aidge Model Adapter

Plugin of the framework Google Model Explorer to visualize Aidge graphs.

Examples

Visualize ONNX file

model-explorer aidge_mobilenetV2.onnx --extensions=aidge_model_explorer

Visualize wrapper method

Note: This function will automatically embed the visualization in a Jupyter Notebook cell if ran in a notebook, otherwise it will simply open a page in your browser.

import aidge_core
import aidge_model_explorer

lstm = aidge_core.LSTM(in_channels=4, hidden_channels=8, seq_length=5)
model = aidge_core.get_connected_graph_view(lstm)

aidge_model_explorer.visualize(model, "LSTM")

Visualize from config

import aidge_core
import aidge_model_explorer
import model_explorer


lstm = aidge_core.LSTM(in_channels=4, hidden_channels=8, seq_length=5)
lstm_model = aidge_core.get_connected_graph_view(lstm)

config = aidge_model_explorer.config()
config.add_graphview(lstm_model, "lstm")
model_explorer.visualize_from_config(config)