"天枢人工智能开源平台"(简称:"天枢平台")是由之江实验室牵头,联合北京一流科技等共同打造的具有自主知识产权的人工智能开源平台。整个平台由一站式AI模型开发平台、高性能深度学习框架和模型炼知框架三大子系统组成。 其中, "一站式AI模型开发平台"面向AI模型生产的生命周期,提供了包括数据处理、模型开发、模型训练和模型管理等功能,方便用户一站式构建AI算法。
原始仓库地址:https://gitee.com/zhijiangtianshu/Dubhe.git
在这个版本中,我们提供了几个基于OneFlow的计算机视觉应用,包括:
- Re-ID
- DCGAN
- Pix2Pix
- SRGAN
- Face Segmentation
- Action Recognition
- Restrack
- TTSR
- AP3D
- GhostNet
- HRNet
- LeNet
- Scnet
- poseNet
- TCA
原始仓库地址:https://gitee.com/zhijiangtianshu/Model_Script_Library.git
KAmalEngine(KAE)的目的是为知识合并和可转移性估计建立一个轻量级的算法包。
特性
- 知识融合和提炼算法
- 易于使用的多任务训练界面
- 基于归因图的深度模型转移性估计
- 预先定义的回调和指标用于评估和可视化
原始仓库地址:https://gitee.com/zhijiangtianshu/Atlas_APP.git
TS-VIS(天枢Vis)是[天枢人工智能开源开放平台](https://gitee.com/zhijiangtianshu/Dubhe)的可视化组件,支持Tensorflow、Pytorch、Oneflow等主流深度学习框架的可视化功能。
**[文档: https://feyily.github.io/tsvis-document/](https://feyily.github.io/tsvis-document/)**

## 亮点
* 框架无关,支持TensorFlow、PyTorch、OneFlow等主流深度学习框架可视化
* 超快的响应速度
* 支持大规模的可视化
* 支持训练过程实时可视化
* 支持降维分析样本可视化
* 支持神经网络异常可视化
## 支持功能
- 模型结构可视化:可视化网络结构,包括计算图和结构图
- 标量数据可视化:可视化包括神经网络`accuary`和`loss`等的标量数据
- 媒体数据可视化:可视化包括图像、文字、音频在内的媒体数据
- 统计分析可视化:可视化神经网络中权重、偏置等的分布
- 降维分析可视化:通过降维算法,可视化任意高维数据
- 超参分析可视化:可视化不同超参数下的神经网络指标
- 异常检测可视化:将神经网络张量数据映射到二维,可视化张量数据统计信息
- 用户定制可视化:可以将所有功能移动到该模块进行可视化
原始仓库地址:https://gitee.com/zhijiangtianshu/zjvis.git
OneFlow是开源的、采用全新架构设计,世界领先的工业级通用深度学习框架。
- 分布式训练全新体验,多机多卡如单机单卡一样简单
- 完美契合一站式平台(k8s + docker)
- 原生支持超大模型
- 近零运行时开销、线性加速比
- 灵活支持多种深度学习编译器
- 自动混合精度
- 中立开放,合作面广
- 持续完善的算子集、模型库
原始仓库地址:https://gitee.com/zhijiangtianshu/oneflow.git
星际争霸的强化学习demo
原始仓库地址:https://gitee.com/zhijiangtianshu/oneflow_-alpha-star.git
我们提供多种基于 OneFlow 的 NLP 模型和 NLP ops,包括:
- TextCNN, Bi-LSTM for text classification
- RoBERT
原始仓库地址:https://gitee.com/zhijiangtianshu/oneflow_nlp_model.git
使用 Oneflow 框架实现 CycleGan。
原始仓库地址:https://gitee.com/zhijiangtianshu/cycle-gan.git
使用 Oneflow 框架实现 Fast Neural Style。
原始仓库地址:https://gitee.com/zhijiangtianshu/fast-neural-style.git
OneFlow 中的对象检测和实例分割神经网络。
原始仓库地址:https://gitee.com/zhijiangtianshu/oneflow_detection.git