环境准备
-
模型转换需要按照概述Atlas 200 DK 开发者套件(1.0.9.alpha)环境部署搭建开发环境华为云 (huaweicloud.com)中的要求为ubuntu18.04的系统配置Ascend环境,本文选择1.0.9.alpha的环境进行开发。
-
需要安装darknet转caffe的开源工具https://github.com/ChenYingpeng/darknet2caffe
操作步骤
-
在darknet2caffe开源工具的文件夹下使用
python darknet2caffe.py cfg[in] weights[in] prototxt[out] caffemodel[out]
将.weights格式的文件转换为caffe格式的文件。 -
按照YOLOv3网络模型prototxt修改_Atlas 200 DK 开发者套件(1.0.9.alpha)_模型构建_通过ATC工具转换模型_定制网络修改(Caffe)_特性配置_样例参考_华为云中的步骤,手动修改.prototxt文件。
-
按照samples: CANN Samples - Gitee.com中的第三步,使用ATC工具将caffe格式的文件转换为能过够部署在Atlas 200上的.om文件。