行李箱物品x光安检项目开发经验分享

BILIBILI视频链接：行李箱安检识别_哔哩哔哩_bilibili

1.MindStudio介绍

MindStudio是一套基于华为自研昇腾AI处理器开发的AI全栈开发工具平台，该IDE上功能很多，涵盖面广，可以进行包括网络模型训练、移植、应用开发、推理运行及自定义算子开发等多种任务。MindStudio除了具有工程管理、编译、调试、运行等一般功能外，还能进行性能分析，算子比对，可以有效提高工作人员的开发效率。除此之外，MindStudio具有远端环境，运行任务在远端实现，对于近端的个人设备的要求不高，用户交互体验很好，可以让我们随时随地进行使用。

2.案例模型

本项目目的是利用MindStudio开发基于Pytorch的训练模型和基于MindX SDK的推理项目。项目对x光安检图象进行推理，利用目标检测框架，对警棍、老虎钳、榔头、充电宝、剪刀、扳手、枪支、子弹、喷罐、手铐、小刀、打火机以上目标进行检测，将检测得到的不同类的目标用不同颜色的矩形框标记。

支持硬件平台：本项目以昇腾Atlas310卡、Atlas 200DK为主要的硬件推理平台。

3.环境搭建

3.1远端环境搭建

在服务器Linux端(aarch64架构)：

1)下载好 MindX SDK mxvision 2.0.4 下载参考链接。

在远端任意用户文件夹下，将Ascend-mindxsdk-mxvision_2.0.4_linux-aarch64.run文件上传

chmod +x Ascend-mindxsdk-mxvision_2.0.4_linux-aarch64.run

./Ascend-mindxsdk-mxvision_3.0.RC2_linux-x86_64.run --install

安装完成后，使环境变量生效：

export DDK_PATH=$HOME/Ascend/ascend-toolkit/latest

export NPU_HOST_LIB=$DDK_PATH/acllib/lib64/stub

source ~/Ascend/ascend-toolkit/set_env.sh

cd ~/software/mxVision/

./set_env.sh

1. 下载好CANN 5.0.4，下载参考链接。

同样在任意用户文件夹下，上传

Ascend-cann-toolkit_5.1.RC2.2_linux-aarch64.run

chmod +x Ascend-cann-toolkit_5.1.RC2.2_linux-aarch64.run

./Ascend-cann-toolkit_5.1.RC2.2_linux-aarch64.run --install

显示如下则CANN 5.0.4安装完成。

安装完成后，可以创建一个env.txt 文件将文档中红框路径保存以下。

3.2 本地（windows端）环境搭建

1)windows版本MindStudio。按照官方步骤进安装，同时安装相应依赖。

2)安装任意版本的Anoconda。

3)按照官方教程进行vpn连接。进入众智实验室容器IP。

4）同步CANN。

打开MindStudio，在File中选择File > Settings > Appearance & Behavior > System Settings > CANN

点击change CANN

进入下图界面，选择远端CANN安装的位置

配置好后返回上一界面显示如图。

 

5）在File > Settings > Appearance & Behavior > System Settings > MindX SDK，进入MindX SDK管理页面，点击Install进入远程配置界面。

在Remote Connection一栏点击加号，配置远程连接，输入申请的账号密码等相关信息。

在Remote CANN Location 一栏选择远端安装CANN Toolkits的路径，具体到版本，这里选择的是5.0.4版本。

在Remote SDK Location 一栏选择远端安装MindX SDK的路径，这里选择的是Mxvision 2.0.4 版本。

Local SDK location为本地默认的SDK安装位置，会将远端的SDK同步过来。

点击ok后开始同步。

配置完成后，界面如下图所示。

4.Pytorch yolox模型训练流程

 

点击File >New >Project 创建一个空训练工程。

选择Pytorch项目。