[自荐] 不写一行代码也能训练AI模型！CortexNodus：把复杂神经网络设计可视化

[streetartist]

项目地址

https://github.com/streetartist/CortexNodus

类别

人工智能

项目标题

用节点图的方式，不写一行代码也能训练AI模型！

项目描述

这是一个基于 Flask + LiteGraph.js 的可视化 AI 训练工作台，支持通过拖拽方式构建神经网络架构并训练。

这个项目极大的拉低了ai训练的门槛，让人快速上手ai模型构建，也能够让ai工程师迅速调整模型架构，同时直观的节点图也能方便教学与研究。

亮点

CortexNodus通过图形化设计器、子图管理与自动代码生成，把模型结构表达、复用与代码实现连接起来：

• 图形化编辑器： 在浏览器中拖拽节点并连线，设计模型的架构

• 子图复用： 支持内联子图与独立子图 ，模块化开发，构建复杂模型也不在话下

• 代码导出： 可以导出可读的 Pytorch 代码，方面二次修改；可以导出小应用的代码，直观展示成果

示例代码

### 快速上手
----



git clone https://github.com/streetartist/CortexNodus.git

cd CortexNodus

pip install -r requirements.txt

# 启动服务器

python app.py

# 打开浏览器访问 http://localhost:5000 开始试玩！



点击打开按钮，选取 example 文件夹中的示例，快速体验 CortexNodus 的强大功能！

截图或演示视频

可编辑的节点参数

实时显示训练数据

实时显示训练日志

支持一键导出APP

图为EfficientNet-B0，支持复杂模型构建！

支持子图