初识新一代开发者套件Atlas 200I DK A2之模型适配工具
模型适配工具是一款集成数据集管理、模型训练、模型打包为一体的开发者工具套件。通过此工具,可以降低开发者在模型开发过程中对AI专业知识、深度学习框架的学习成本,极大降低开发的难度、复杂度。
1.功能介绍
- 工具预置了4个典型模型,分别用于目标检测场景(YOLO V5模型)、图像分类场景(Mobilenet V3模型)、图像分割场景(Unet ++模型)、关键点检测场景(Alphapose模型)。
- 提供UI操作界面,开发者可在界面上传自己的数据集,工具支持对数据进行标注和转换。
- 支持用户在工具界面创建模型训练任务,并使用PC的CPU算力进行训练。
- 支持对训练完成模型进行打包,便于拷贝到开发者套件中。
2.安装流程
a. 安装Anaconda3并在安装模型适配工具前准备好虚拟环境
b.获取并安装模型适配工具
c.模型适配工具标注工具制作数据集
d.本机模型训练与迁移
e.上传模型迁移压缩包到DK上,进行模型ATC转换om模型和推理主程序
标签:DK,CANN,适配,模型,套件,开发者,工具 From: https://blog.51cto.com/u_14125503/6378048