一、学习目标
1. 掌握如何快速运行开箱样例
2.掌握如何使用一键制卡工具
3.掌握如何使用模型适配工具
二、一键制卡工具
1.介绍
在使用Atlas 200l DKA2进行开发时,需要SD卡作为储存设备,开发板也需要多种依赖和驱动,一键制卡工具可以在线/本地将适配好开发板的镜像(镜像中包含OS、NPU固件驱动、CANN、代码样例等)烧录到SD卡。
2.相关准备
(1)硬件
- 开发者套件,包括套件主板和电源
- Micro SD卡和读卡器
- 连接线:Type-C数据线或RJ45网线
- PC(笔记本或台式机)
(2)软件
制卡工具,全称ascend-ai-devkit-imager,用于烧录镜像到SD卡。
3.过程
目前提供2种镜像:基础镜像与E2E镜像。基础镜像只包含基础样例(如图像分类、目标检测),E2E镜像包含基础样例和E2E样例(智能车、机械臂、聊天机器人等)。
三、开箱样例
开发者套件上电后,D3、LED1、LED3指示灯会依次绿色常亮,表示启动正常。
使用快捷键“Win+R”,在运行窗口输入cmd进入命令行窗口。输入ipconfig命令查询PC网口IP地址是否修改成功。
root用户密码默认为Mind@123
四、模型适配工具
1.介绍
模型适配工具是一款集成数据集管理、模型训练、模型打包为一体的开发者工具套件。通过此工具,可以降低开发者在模型开发过程中对AI专业知识、深度学习框架的学习成本,极大降低开发的难度、复杂度。
(1)工具预置了4个典型模型,分别用于目标检测场景(YOLO V5模型)、图像分类场景(Mobilenet V3模型)、图像分割场景(Unet ++模型)、关键点检测场景(Alphapose模型)。
(2)提供UI操作界面,开发者可在界面上传自己的数据集,工具支持对数据进行标注和转换。
(3)支持用户在工具界面创建模型训练任务,并使用PC的CPU算力进行训练。
(4)支持对训练完成模型进行打包,便于拷贝到开发者套件中。
2.环境配置及软件安装
3.构件图像分类推理应用
(1)制作数据集
为模型迁移准备数据集,进行图像标注,在工具界面选择“分类模型”。
选择“制作数据集”,配置数据集的路径和标签。单击“确认”,开始标注,对应每张图片,从右侧标记栏选择对应的标签名称,当前图片标注完成后,从图片上方菜单单击图标或左侧文件列表选择下一张图片进行标记,直到完成所有图片的标注任务。
(2)模型迁移
在工具界面单击下方“一键迁移”按钮,进入配置界面,输入迁移信息。点击一键迁移。
用户可以在模型适配工具后台命令行窗口查看迁移过程数据
迁移完成后会出现提示框,提示已生成打包好的文件,在训练输出目录会生成以下文件与目录:
train_output:训练输出的权重文件、onnx文件以及训练数据信息json文件。
trans_output:经过数据转换,根据数据集拆分设置生成的测试集、验证集、训练集。
edge_infer.tar:打包好的推理相关模型文件与脚本。
(3)模型推理
将(2)中输出的压缩包发送到开发板上并解压,进入解压后的路径下执行以下命令将Windows格式文件转换为Unix、Linux格式。
cd ~/edge_infer/ && dos2unix `find .`
将.onnx格式模型转换适用于昇腾AI处理器的om格式模型文件。
bash atc.sh
执行以下命令进行推理。
bash run.sh