├── .github └── workflows │ ├── pyinstaller-build.yaml │ └── test_digest.yaml ├── .gitignore ├── .pylintrc ├── LICENSE.txt ├── README.md ├── examples ├── README.md └── analysis.py ├── main.spec ├── pyproject.toml ├── setup.py ├── src ├── digest │ ├── __init__.py │ ├── assets │ │ ├── gifs │ │ │ ├── digest.gif │ │ │ └── load.gif │ │ ├── icons │ │ │ ├── close-window-64.ico │ │ │ ├── digest_logo.ico │ │ │ ├── file.png │ │ │ ├── freeze.png │ │ │ ├── huggingface.png │ │ │ ├── huggingface_64px.png │ │ │ ├── info.png │ │ │ ├── models.png │ │ │ ├── node_list.png │ │ │ ├── open.png │ │ │ ├── save.png │ │ │ ├── search.png │ │ │ └── summary.png │ │ └── images │ │ │ ├── banner.png │ │ │ ├── digest_logo_500.jpg │ │ │ ├── remove_background_200_zoom.png │ │ │ └── remove_background_500_zoom.png │ ├── clickablelabel.py │ ├── compile_digest_gui.py │ ├── dialog.py │ ├── freeze_inputs.py │ ├── gui_config.yaml │ ├── histogramchartwidget.py │ ├── huggingface_page.py │ ├── main.py │ ├── model_class │ │ ├── digest_model.py │ │ ├── digest_onnx_model.py │ │ └── digest_report_model.py │ ├── modelsummary.py │ ├── multi_model_analysis.py │ ├── multi_model_selection_page.py │ ├── node_summary.py │ ├── piechartwidget.py │ ├── plaintexteditentersignal.py │ ├── popup_window.py │ ├── qt_utils.py │ ├── resource.qrc │ ├── resource_rc.py │ ├── similarity_analysis.py │ ├── styles │ │ └── darkstyle.qss │ ├── subgraph_analysis │ │ ├── __init__.py │ │ ├── database.zip │ │ ├── find_match.py │ │ └── model_encode.py │ └── ui │ │ ├── freezeinputs.ui │ │ ├── freezeinputs_ui.py │ │ ├── huggingface_page.ui │ │ ├── huggingface_page_ui.py │ │ ├── mainwindow.ui │ │ ├── mainwindow_ui.py │ │ ├── modelsummary.ui │ │ ├── modelsummary_ui.py │ │ ├── multimodelanalysis.ui │ │ ├── multimodelanalysis_ui.py │ │ ├── multimodelselection_page.ui │ │ ├── multimodelselection_page_ui.py │ │ ├── nodessummary.ui │ │ └── nodessummary_ui.py └── utils │ ├── __init__.py │ └── onnx_utils.py └── test ├── __init__.py ├── resnet18.onnx ├── resnet18_reports ├── resnet18_heatmap.png ├── resnet18_histogram.png ├── resnet18_node_type_counts.csv ├── resnet18_nodes.csv ├── resnet18_report.txt └── resnet18_report.yaml ├── test_gui.py └── test_reports.py /.github/workflows/pyinstaller-build.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/digestai/HEAD/.github/workflows/pyinstaller-build.yaml -------------------------------------------------------------------------------- /.github/workflows/test_digest.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/digestai/HEAD/.github/workflows/test_digest.yaml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/digestai/HEAD/.gitignore -------------------------------------------------------------------------------- /.pylintrc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/digestai/HEAD/.pylintrc -------------------------------------------------------------------------------- /LICENSE.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/digestai/HEAD/LICENSE.txt -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/digestai/HEAD/README.md -------------------------------------------------------------------------------- /examples/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/digestai/HEAD/examples/README.md -------------------------------------------------------------------------------- /examples/analysis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/digestai/HEAD/examples/analysis.py -------------------------------------------------------------------------------- /main.spec: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/digestai/HEAD/main.spec -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/digestai/HEAD/pyproject.toml -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/digestai/HEAD/setup.py -------------------------------------------------------------------------------- /src/digest/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/digest/assets/gifs/digest.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/digestai/HEAD/src/digest/assets/gifs/digest.gif -------------------------------------------------------------------------------- /src/digest/assets/gifs/load.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/digestai/HEAD/src/digest/assets/gifs/load.gif -------------------------------------------------------------------------------- /src/digest/assets/icons/close-window-64.ico: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/digestai/HEAD/src/digest/assets/icons/close-window-64.ico -------------------------------------------------------------------------------- /src/digest/assets/icons/digest_logo.ico: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/digestai/HEAD/src/digest/assets/icons/digest_logo.ico -------------------------------------------------------------------------------- /src/digest/assets/icons/file.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/digestai/HEAD/src/digest/assets/icons/file.png -------------------------------------------------------------------------------- /src/digest/assets/icons/freeze.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/digestai/HEAD/src/digest/assets/icons/freeze.png -------------------------------------------------------------------------------- /src/digest/assets/icons/huggingface.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/digestai/HEAD/src/digest/assets/icons/huggingface.png -------------------------------------------------------------------------------- /src/digest/assets/icons/huggingface_64px.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/digestai/HEAD/src/digest/assets/icons/huggingface_64px.png -------------------------------------------------------------------------------- /src/digest/assets/icons/info.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/digestai/HEAD/src/digest/assets/icons/info.png -------------------------------------------------------------------------------- /src/digest/assets/icons/models.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/digestai/HEAD/src/digest/assets/icons/models.png -------------------------------------------------------------------------------- /src/digest/assets/icons/node_list.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/digestai/HEAD/src/digest/assets/icons/node_list.png -------------------------------------------------------------------------------- /src/digest/assets/icons/open.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/digestai/HEAD/src/digest/assets/icons/open.png -------------------------------------------------------------------------------- /src/digest/assets/icons/save.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/digestai/HEAD/src/digest/assets/icons/save.png -------------------------------------------------------------------------------- /src/digest/assets/icons/search.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/digestai/HEAD/src/digest/assets/icons/search.png -------------------------------------------------------------------------------- /src/digest/assets/icons/summary.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/digestai/HEAD/src/digest/assets/icons/summary.png -------------------------------------------------------------------------------- /src/digest/assets/images/banner.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/digestai/HEAD/src/digest/assets/images/banner.png -------------------------------------------------------------------------------- /src/digest/assets/images/digest_logo_500.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/digestai/HEAD/src/digest/assets/images/digest_logo_500.jpg -------------------------------------------------------------------------------- /src/digest/assets/images/remove_background_200_zoom.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/digestai/HEAD/src/digest/assets/images/remove_background_200_zoom.png -------------------------------------------------------------------------------- /src/digest/assets/images/remove_background_500_zoom.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/digestai/HEAD/src/digest/assets/images/remove_background_500_zoom.png -------------------------------------------------------------------------------- /src/digest/clickablelabel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/digestai/HEAD/src/digest/clickablelabel.py -------------------------------------------------------------------------------- /src/digest/compile_digest_gui.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/digestai/HEAD/src/digest/compile_digest_gui.py -------------------------------------------------------------------------------- /src/digest/dialog.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/digestai/HEAD/src/digest/dialog.py -------------------------------------------------------------------------------- /src/digest/freeze_inputs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/digestai/HEAD/src/digest/freeze_inputs.py -------------------------------------------------------------------------------- /src/digest/gui_config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/digestai/HEAD/src/digest/gui_config.yaml -------------------------------------------------------------------------------- /src/digest/histogramchartwidget.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/digestai/HEAD/src/digest/histogramchartwidget.py -------------------------------------------------------------------------------- /src/digest/huggingface_page.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/digestai/HEAD/src/digest/huggingface_page.py -------------------------------------------------------------------------------- /src/digest/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/digestai/HEAD/src/digest/main.py -------------------------------------------------------------------------------- /src/digest/model_class/digest_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/digestai/HEAD/src/digest/model_class/digest_model.py -------------------------------------------------------------------------------- /src/digest/model_class/digest_onnx_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/digestai/HEAD/src/digest/model_class/digest_onnx_model.py -------------------------------------------------------------------------------- /src/digest/model_class/digest_report_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/digestai/HEAD/src/digest/model_class/digest_report_model.py -------------------------------------------------------------------------------- /src/digest/modelsummary.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/digestai/HEAD/src/digest/modelsummary.py -------------------------------------------------------------------------------- /src/digest/multi_model_analysis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/digestai/HEAD/src/digest/multi_model_analysis.py -------------------------------------------------------------------------------- /src/digest/multi_model_selection_page.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/digestai/HEAD/src/digest/multi_model_selection_page.py -------------------------------------------------------------------------------- /src/digest/node_summary.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/digestai/HEAD/src/digest/node_summary.py -------------------------------------------------------------------------------- /src/digest/piechartwidget.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/digestai/HEAD/src/digest/piechartwidget.py -------------------------------------------------------------------------------- /src/digest/plaintexteditentersignal.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/digestai/HEAD/src/digest/plaintexteditentersignal.py -------------------------------------------------------------------------------- /src/digest/popup_window.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/digestai/HEAD/src/digest/popup_window.py -------------------------------------------------------------------------------- /src/digest/qt_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/digestai/HEAD/src/digest/qt_utils.py -------------------------------------------------------------------------------- /src/digest/resource.qrc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/digestai/HEAD/src/digest/resource.qrc -------------------------------------------------------------------------------- /src/digest/resource_rc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/digestai/HEAD/src/digest/resource_rc.py -------------------------------------------------------------------------------- /src/digest/similarity_analysis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/digestai/HEAD/src/digest/similarity_analysis.py -------------------------------------------------------------------------------- /src/digest/styles/darkstyle.qss: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/digestai/HEAD/src/digest/styles/darkstyle.qss -------------------------------------------------------------------------------- /src/digest/subgraph_analysis/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/digest/subgraph_analysis/database.zip: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/digestai/HEAD/src/digest/subgraph_analysis/database.zip -------------------------------------------------------------------------------- /src/digest/subgraph_analysis/find_match.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/digestai/HEAD/src/digest/subgraph_analysis/find_match.py -------------------------------------------------------------------------------- /src/digest/subgraph_analysis/model_encode.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/digestai/HEAD/src/digest/subgraph_analysis/model_encode.py -------------------------------------------------------------------------------- /src/digest/ui/freezeinputs.ui: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/digestai/HEAD/src/digest/ui/freezeinputs.ui -------------------------------------------------------------------------------- /src/digest/ui/freezeinputs_ui.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/digestai/HEAD/src/digest/ui/freezeinputs_ui.py -------------------------------------------------------------------------------- /src/digest/ui/huggingface_page.ui: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/digestai/HEAD/src/digest/ui/huggingface_page.ui -------------------------------------------------------------------------------- /src/digest/ui/huggingface_page_ui.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/digestai/HEAD/src/digest/ui/huggingface_page_ui.py -------------------------------------------------------------------------------- /src/digest/ui/mainwindow.ui: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/digestai/HEAD/src/digest/ui/mainwindow.ui -------------------------------------------------------------------------------- /src/digest/ui/mainwindow_ui.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/digestai/HEAD/src/digest/ui/mainwindow_ui.py -------------------------------------------------------------------------------- /src/digest/ui/modelsummary.ui: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/digestai/HEAD/src/digest/ui/modelsummary.ui -------------------------------------------------------------------------------- /src/digest/ui/modelsummary_ui.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/digestai/HEAD/src/digest/ui/modelsummary_ui.py -------------------------------------------------------------------------------- /src/digest/ui/multimodelanalysis.ui: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/digestai/HEAD/src/digest/ui/multimodelanalysis.ui -------------------------------------------------------------------------------- /src/digest/ui/multimodelanalysis_ui.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/digestai/HEAD/src/digest/ui/multimodelanalysis_ui.py -------------------------------------------------------------------------------- /src/digest/ui/multimodelselection_page.ui: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/digestai/HEAD/src/digest/ui/multimodelselection_page.ui -------------------------------------------------------------------------------- /src/digest/ui/multimodelselection_page_ui.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/digestai/HEAD/src/digest/ui/multimodelselection_page_ui.py -------------------------------------------------------------------------------- /src/digest/ui/nodessummary.ui: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/digestai/HEAD/src/digest/ui/nodessummary.ui -------------------------------------------------------------------------------- /src/digest/ui/nodessummary_ui.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/digestai/HEAD/src/digest/ui/nodessummary_ui.py -------------------------------------------------------------------------------- /src/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/utils/onnx_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/digestai/HEAD/src/utils/onnx_utils.py -------------------------------------------------------------------------------- /test/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/resnet18.onnx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/digestai/HEAD/test/resnet18.onnx -------------------------------------------------------------------------------- /test/resnet18_reports/resnet18_heatmap.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/digestai/HEAD/test/resnet18_reports/resnet18_heatmap.png -------------------------------------------------------------------------------- /test/resnet18_reports/resnet18_histogram.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/digestai/HEAD/test/resnet18_reports/resnet18_histogram.png -------------------------------------------------------------------------------- /test/resnet18_reports/resnet18_node_type_counts.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/digestai/HEAD/test/resnet18_reports/resnet18_node_type_counts.csv -------------------------------------------------------------------------------- /test/resnet18_reports/resnet18_nodes.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/digestai/HEAD/test/resnet18_reports/resnet18_nodes.csv -------------------------------------------------------------------------------- /test/resnet18_reports/resnet18_report.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/digestai/HEAD/test/resnet18_reports/resnet18_report.txt -------------------------------------------------------------------------------- /test/resnet18_reports/resnet18_report.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/digestai/HEAD/test/resnet18_reports/resnet18_report.yaml -------------------------------------------------------------------------------- /test/test_gui.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/digestai/HEAD/test/test_gui.py -------------------------------------------------------------------------------- /test/test_reports.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/digestai/HEAD/test/test_reports.py --------------------------------------------------------------------------------