nnie学习一、生成wk文件

目前,nnie只支持caffemodel,windows上有两种方法生成wk文件,第一种是cmd里敲命令,第二种是利用ruyistudio。

cmd里敲命令的步骤如下:

第一步、准备好deploy prototxt、caffemodel、图片等;

第二步、写cfg文件,如:

nnie学习一、生成wk文件_第1张图片

其中,instruction_name指定输出的wk文件保存路径和名字。

第三步、cd到海思的sample_simulator,

执行如下命令

nnie学习一、生成wk文件_第2张图片

则会生成wk文件和cnn_net_tree.dot文件,cnn_net_tree.dot即为标记后的网络结构文件。

到此wk文件生成完成。

想看网络结构的话,执行如下命令将dot文件保存成pdf或者jpg进行查看,

网络结构如下所示:

nnie学习一、生成wk文件_第3张图片

你可能感兴趣的:(caffe,nnie,wk文件生成,海思)