海思hi3516dv300芯片AI图像模块学习（六）Caffe模型量化

2022-07-27 18:35 作者:饼干快快快跑 0人读过 | 我要投稿

据说Caffe模型转wk模型的精度损失最小，所以不管什么模型都要先转Caffe在导wk。而darknet框架据说与Caffe框架是最相似的，两者转化需要替换的层较少，所以官方推荐了darknet的yolov3转caffe。但是darknet框架使用的人较少，对yolo高版本的更新较慢。

hisi通过NNIE对神经网络进行硬件加速，要调用NNIE需要将网络模型转化成在Hi3516芯片上可以运行的wk文件。海思官方通过nnie_mapper工具进行转换。这个工具被集成到Windows版的IDE工具RuyiStudio上。但是就像darknet到Caffe时需要替换或增加部分层一样，由于NNIE的限制Caffe模型中的部分层也不能在NNIE中调用，需要提前对caffee模型中的.prototxt文件提前修改。

1.RuyiStudio 安装

1.1 编译链MinGW-W64 安装

参考：https://blog.csdn.net/xm961217/article/details/105637100

下载wget安装包，根据猜测，wget 是用于快捷搭建环境的一个小工具。然后，将刚刚安装的目录下的 wget.exe 所在的目录添加到环境变量中，也就是我的电脑右键 -> 属性 -> 高级系统设置 -> 环境变量

运行点击ruyi_env_setup文件夹（这个文件夹不能在wget里面）下的setup_mingw.bat

setup_mingw.bat自动编译成功。

1.2 Python3.5+caffe 环境配置

点击setup_python.bat进行安装，

报错

Failed to extract wheel-0.29.0-py35_0.tar.bz2, please check from the requirements.txt, download and extract manually

Failed to extract zlib-1.2.11-vc14_0.tar.bz2, please check from the requirements.txt, download and extract manually;

Failed at installing opencv_python-3.4.2.16-cp35-cp35m-win_amd64.whl, please install it by "pip install opencv_python-3.4.2.16-cp35-cp35m-win_amd64.whl" command under python35 directory;

Failed at installing protobuf-3.6.1-cp35-cp35m-win_amd64.whl, please install it by "pip install protobuf-3.6.1-cp35-cp35m-win_amd64.whl" command under python35 directory;

Failed at installing PyYAML-3.13-cp35-cp35m-win_amd64.whl, please install it by "pip install PyYAML-3.13-cp35-cp35m-win_amd64.whl" command under python35 directory;

Failed at installing Cython-0.28.5-cp35-cp35m-win_amd64.whl, please install it by "pip install Cython-0.28.5-cp35-cp35m-win_amd64.whl" command under python35 directory;

Failed at extracting, please check "D:\RuyiStudioSetup\ruyi_env_setup-2.0.41\setup.log".

pip报错：

pip install opencv_python-3.4.2.16-cp35-cp35m-win_amd64.whl

python -m pip install --upgrade pip -i https://pypi.douban.com/simple

用国内镜像更新pip ，解决。

再点击setup_python.bat进行安装，

继续报错

Start setup_extract.bat

Failed to extract zlib-1.2.11-vc14_0.tar.bz2, please check from the requirements.txt, download and extract manually;

Failed at extracting, please check "D:\RuyiStudioSetup\ruyi_env_setup-2.0.41\setup.log".

End setup_extract.bat.

zlib没有装上，手动装一下