NNDesigner

NNDesigner 是一个基于 PyTorch 的可视化深度学习模型设计工具，提供直观的图形界面来构建和管理神经网络模型。

功能特点

• 基于 PySide6 的现代化图形界面
• 支持拖拽式神经网络设计
• 多标签页模型管理
• 深色主题支持
• 节点搜索功能
• 可扩展的节点系统

环境要求

• Python >= 3.7
• PySide6
• NodeGraphQt
• PyTorch
• qdarkstyle (可选，用于界面美化)

安装

pip install nndesigner

使用方法

nndesigner

开发

git clone https://github.com/xiaoqiang-cheng/nndesigner.git
cd nndesigner
pip install -e .

NNDesigner (English)

NNDesigner is a visual deep learning model design tool based on PyTorch, providing an intuitive graphical interface for building and managing neural network models.

Features

• Modern GUI based on PySide6
• Drag-and-drop neural network design
• Multi-tab model management
• Dark theme support
• Node search functionality
• Extensible node system

Requirements

• Python >= 3.7
• PySide6
• NodeGraphQt
• PyTorch
• qdarkstyle (optional, for UI styling)

Installation

pip install nndesigner

Usage

nndesigner

Development

git clone https://github.com/xiaoqiang-cheng/nndesigner.git
cd nndesigner
pip install -e .

reference

https://chantonic.com/NodeGraphQt/api/examples/ex_overview.html