【CANN训练营第三季】特邀直播openmmlab
·

1、编写mmdeploy在异腾环境下的安装脚本
pip install torch==1.8.1 torchvision==0.9.1 --extra-index-url https://download.pytorch.org/whl/cpu
pip install openmim
mim install mmcv-full
sudo apt-get install libopencv-dev
git clone --recursive https://github.com/open-mmlab/mmdeploy.git
cd mmdeploy
#安装 mmdeploy 楼型转换器
pip install -v -e .
#编译 mmdeploy 推理 SDK
## 设置 ascend-toolkit 的环境变量
#source /usr/local/Ascend/ascend-toolkit/set_env.sh
source /home/HwHiAiUser/Ascend/ascend-toolkit/set_env.sh
mkdir -p build && cd build
cmake .. -DMMDEPLOY_BUILD_SDK=ON -DMMDEPLOY_BUILD_SDK_PYTHONAPI=ON -DMMDEPLOY_TARGET_BACKEND=acl
make -j$(nproc) && make install
cd ..
2、编写测试脚本,验证resnet50、retinanet适配cann的转换和推理过程
#安装 openmmlab 算法库 mmcls
pip install mmcls
# 使用 mim 下载 mmcls 中 resnet18 的模型和 ckpt
mim download mmcls --config resnet18_8xb32_in1k --dest .
#执行 mmdeploy 的转换命令,得到 om 模型以及 mmdeploy model
python tools/deploy.py \
configs/mmcls/classification_ascend_static-224x224.py \
resnet18_8xb32_in1k.py \
resnet18_8xb32_in1k_20210831-fbbb1da6.pth \
tests/data/tiger.jpeg \
--work-dir mmdeploy_models/mmcls/resnet18/cann \
--device cpu\
--dump-info
fast-rnn
#安装 openmmlab 算法库 mmdet
pip install mmdet
#使用mim 下载 mmdet 中 faster r-cnn 的模型和ckpt
mim download mmdet --config faster_rcnn_r50_fpn_1x_coco --dest .
#执行 mmdeploy 的转换命令,得到 .om 模型以及 mmdeploy model
python tools/deploy.py \
configs/mmdet/detection/detection_ascend_static-800x1344.py \
faster_rcnn_r50_fpn_1x_coco.py \
faster_rcnn_r5e_fpn_1x_coco20200130-047c8118.pth \
demo/resources/det.jpg \
--work-dir mmdeploy_models/mmdet/faster-rcnn/cann \
--device cpu \
--dump-info
3、升级到 CANN 6.0,验证 mmdeploy 部署功能的正确性
直接在cann6.0测试即可
昇腾计算产业是基于昇腾系列(HUAWEI Ascend)处理器和基础软件构建的全栈 AI计算基础设施、行业应用及服务,https://devpress.csdn.net/organization/setting/general/146749包括昇腾系列处理器、系列硬件、CANN、AI计算框架、应用使能、开发工具链、管理运维工具、行业应用及服务等全产业链
更多推荐

所有评论(0)