生成wk文件后,进行中间层结果对比验证.
Ruyi Studio提供了Vector Comparision工具,能够对比输出向量的相似度、绝对误差等信息,可用于验证模型量化后的输出精度误差大小。
同时海思提供了一个中间层输出脚本,其路径如下。该工具可以通过读取cfg文件输出中间层结果。
HiSVP_PC_V1.1.2.0/tools/nnie/windows/RuyiStudio-2.0.31/Resources/pythonScript/CNN_convert_bin_and_print_featuremap.py
将该脚本复制到工程的data目录下,同时复制一份MobileFace.cfg文件到data目录下,脚本运行时需要读取cfg文件。
此处的cfg文件有两处需要修改的地方。删除image_list选项,增加image_file选项,路径为量化时使用的image_list文件的最后一张图片的路径。(-代表删除,+代表添加)
- [image_list] ./data/images/imageList.txt
+ [image_file] ./images/9.jpg
修改mean_file路径为
- [mean_file] ./data/pixel_mean.txt
+ [mean_file] ./pixel_mean.txt
执行脚本:
python CNN_convert_bin_and_print_featuremap.py -i MobileFace.cfg -m mobilefacenet.prototxt -w mobilefacenet.caffemodel -c 0
出现问题:
一、ImportError: No module named 'skimage'
解决方法:pip install scikit-image
二、ImportError: No module named 'scipy'
解决方法:pip install -U scipy
三、ImportError: No module named 'numpy'
conda install numpy
四、
五、