主要内容

基于ARM计算的深度学习预测codegen

这个例子展示了如何使用codegen为在ARM®处理器上使用深度学习的Logo分类应用程序生成代码。标志分类应用程序使用LogoNet系列网络从图像中执行标识识别。生成的代码利用ARM Compute库进行计算机视觉和机器学习。

先决条件

  • 支持NEON扩展的ARM处理器

  • 开源计算机视觉库(OpenCV) v3.1

  • ARM Compute和OpenCV库的环境变量

  • 用于c++代码生成的MATLAB®Coder™

  • 深度学习的支持包MATLAB编码器接口

  • 深度学习工具箱™用于使用SeriesNetwork对象

本例使用的ARM Compute库版本可能不是代码生成支持的最新版本。有关受支持的库版本和有关设置环境变量的信息,请参见用MATLAB编码器进行深度学习的先决条件

此示例在Linux®和Windows®平台上支持,而在MATLAB Online上不支持。

获得预培训系列网络

下载pretrainedLogoNet联网并保存为logonet.mat,如果它不存在。该网络是在MATLAB®中开发的,其体系结构与AlexNet类似。该网络可以在不同的照明条件和相机角度下识别32个logo。

网= getLogonet ();

该网络包含22层,包括卷积层、全连接层和分类输出层。

网层
ans = 22×1带有图层的图层数组:1“imageinput”图像输入227×227×3图片zerocenter正常化和randfliplr的扩增2 conv_1卷积96 5×5×3旋转步[1]和填充[0 0 0 0]3‘relu_1 ReLU ReLU 4“maxpool_1”马克斯池3×3马克斯池步(2 - 2)和填充[0 0 0 0]5 conv_2卷积128 3×3×96旋转步[1]和填充[0 0 0 0]6‘relu_2 ReLU ReLU 7“maxpool_2”马克斯池3×3马克斯池步(2 - 2)和填充[0 0 0 0]8“conv_3”卷积384 3×3×128旋转步[1]和填充[0 0 0 0]9 ' relu_3 ReLU ReLU 10“maxpool_3”马克斯池3×3马克斯池步(2 - 2)和填充[0 0 0 0]11 conv_4卷积128 3×3×384旋转步(2 - 2)和填充[0 0 0 0]12的relu_4 ReLU ReLU 13“maxpool_4”马克斯池3×3马克斯池步(2 - 2)和填充[0 0 0 0]14 fc_1完全连接2048完全连接层15 ' relu_5 ReLU ReLU 16“dropout_1”辍学50%辍学17 fc_2完全Connected 2048 fully connected layer 18 'relu_6' ReLU ReLU 19 'dropout_2' Dropout 50% dropout 20 'fc_3' Fully Connected 32 fully connected layer 21 'softmax' Softmax softmax 22 'classoutput' Classification Output crossentropyex with 'adidas' and 31 other classes

设置环境变量

在ARM目标硬件上,确保设置了ARM_COMPUTELIB,并且LD_LIBRARY_PATH包含ARM计算库文件夹的路径。

看到用MATLAB编码器进行深度学习的先决条件

logonet_predict函数

logonet_predict.m入口点函数接受图像输入,并使用保存在的深度学习网络对图像进行预测LogoNetMAT-file。函数从中加载网络对象LogoNet.mat变成一个持久的网络变量logonet.在对该函数的后续调用中,将重用持久对象。

类型logonet_predict
function out = logonet_predict(in) %#codegen %版权所有if isempty(logonet) logonet = code . loaddeeplearningnetwork (' logonet .mat','logonet');End out = logonet.predict(in);结束

为静态库设置代码生成配置对象

当您生成针对基于arm的设备而不使用硬件支持包的代码时,请为库创建一个配置对象。不要为可执行程序创建配置对象。

为生成c++代码和仅生成代码设置配置对象。

cfg = coder.config (“自由”);cfg。TargetLang =“c++”;cfg。GenCodeOnly = true;

为深度学习代码生成设置配置对象

创建一个编码器。ARMNEONConfig对象。指定目标ARM处理器的库版本和体系结构。例如,假设目标板是HiKey/Rock960板,具有ARMv8架构,ARM计算库版本为19.05。

dlcfg =编码器。DeepLearningConfig (“arm-compute”);dlcfg。ArmComputeVersion =“19.05”;dlcfg。ArmArchitecture =“armv8”

将深度学习配置对象附加到代码生成配置对象

设置DeepLearningConfig属性为深度学习配置对象的代码生成配置对象。

cfg。DeepLearningConfig = dlcfg;

生成源c++代码codegen

codegen配置cfglogonet_predictarg游戏{ones(227, 227, 3, 'single')}- darm_compute

生成的代码arm_compute文件夹在主机上的当前工作文件夹。

生成Zip文件使用packNGo函数

packNGo函数将所有相关文件打包到一个压缩的zip文件中。

zipFileName =“arm_compute.zip”;bInfo =负载(fullfile (' arm_compute ', ' buildInfo.mat '));packNGo (bInfo。buildInfo, {'fileName', zipFileName,'minimalHeaders', false, 'ignoreFileMissing',true});

将生成的Zip文件复制到目标硬件

复制Zip文件并解压到一个文件夹中。从目标硬件中删除Zip文件。

在以下命令中替换:

  • 密码用你的密码

  • 用户名用你的用户名

  • targetname与您的设备的名称

  • targetloc使用文件的目标文件夹

运行这些命令从Linux中复制并解压zip文件。

如果isunix、系统(['sshpass -p密码SCP -r 'zipFileName fullfile (pwd)“username@targetname: targetloc /”]),结束如果isunix系统('sshpass -p password SSH username@targetname "if [-d targetloc/arm_compute];然后rm -rf targetloc/arm_compute;fi”),结束如果isunix、系统(['sshpass -p password SSH username@targetname "unzip targetloc/'zipFileName“- d targetloc / arm_compute””]),结束如果isunix、系统(['sshpass -p password SSH username@targetname "rm -rf targetloc'zipFileName“””]),结束

运行这些命令从Windows中复制并解压zip文件。

如果ispc、系统(['pscp.exe -pw密码-r 'zipFileName fullfile (pwd)“username@targetname: targetloc /”]),结束如果ispc系统('plink.exe -l username -pw password targetname "if [-d targetloc/arm_compute];然后rm -rf targetloc/arm_compute;fi”),结束如果ispc、系统(['plink.exe -l username -pw password targetname '解压缩targetloc/'zipFileName“- d targetloc / arm_compute””]),结束如果ispc、系统(['plink.exe -l username -pw password targetname ' rm -rf targetloc'zipFileName“””]),结束

复制示例文件到目标硬件

将这些支持文件从主机复制到目标硬件:

  • 输入图像,coderdemo_google.png

  • 生成库的Makefile,logonet_predict_rtw.mk

  • Makefile用于构建可执行程序,makefile_arm_logo.mk

  • 同义词集词典,synsetWordsLogoDet.txt

在以下命令中替换:

  • 密码用你的密码

  • 用户名用你的用户名

  • targetname与您的设备的名称

  • targetloc使用文件的目标文件夹

在从Linux运行时,执行以下步骤复制所有必需的文件

如果isunix系统('sshpass -p password SCP logonet_predict_rtw. '可username@targetname: targetloc / arm_compute /),结束如果isunix系统('sshpass -p密码SCP coderdemo_google.png username@targetname:targetloc/arm_compute/'),结束如果isunix系统('sshpass -p password SCP makefile_arm_logo. '可username@targetname: targetloc / arm_compute /),结束如果isunix系统('sshpass -p password scp synsetwordslogodt .txt username@targetname:targetloc/arm_compute/'),结束

从Windows运行时,执行以下步骤复制所有所需的文件

如果ispc系统('pscp.exe -pw password logonet_predict_rtw. '可username@targetname: targetloc / arm_compute /),结束如果ispc系统('pscp.exe -pw password coderdemo_google.png username@targetname:targetloc/arm_compute/'),结束如果ispc系统(“pscp.exe -pw password makefile_arm_logo. exe”可username@targetname: targetloc / arm_compute /),结束如果ispc系统('pscp.exe -pw password synsetwordslogodt .txt username@targetname:targetloc/arm_compute/'),结束

在目标硬件上构建库

要在目标硬件上构建库,请在ARM硬件上执行生成的makefile。

确保在目标硬件上设置了环境变量ARM_COMPUTELIB和LD_LIBRARY_PATH。看到用MATLAB编码器进行深度学习的先决条件.ARM_ARCH变量在Makefile中用于传递基于Arm架构的编译器标志。在Makefile中使用ARM_VER变量编译基于Arm Compute Version的代码。替换这些命令中的硬件凭据和路径,与上一节类似。

执行以下步骤从Linux构建库。

如果isunix系统('sshpass -p密码SCP main_arm_logo.cpp username@targetname:targetloc/arm_compute/'),结束如果isunix、系统(['sshpass -p password ssh username@targetname "make -C targetloc/arm_compute/ -f logonet_predict_rtw. '可ARM_ARCH = 'dlcfg。ArmArchitecture“ARM_VER = 'dlcfg。ArmComputeVersion“””]),结束

执行以下步骤从窗口构建库。

如果ispc系统('pscp.exe -pw password main_arm_logo.cpp username@targetname:targetloc/arm_compute/'),结束如果ispc、系统(['plink.exe -l username -pw password targetname "make -C targetloc/arm_compute/ -f logonet_predict_rtw. '可ARM_ARCH = 'dlcfg。ArmArchitecture“ARM_VER = 'dlcfg。ArmComputeVersion“””]),结束

从目标硬件上的库中创建可执行文件

使用源主包装器文件构建库,以创建可执行文件。main_arm_logo.cpp是c++主包装器文件,它调用logonet_predict函数。

运行以下命令从Linux创建可执行文件。

如果isunix系统('sshpass -p password ssh username@targetname "make -C targetloc/arm_compute/ -f makefile_arm_logo. '可targetDirName = targetloc / arm_compute”),结束

运行以下命令从Windows创建可执行文件。

如果ispc系统('plink.exe -l username -pw password targetname "make -C targetloc/arm_compute/ -f makefile_arm_logo。可targetDirName = targetloc / arm_compute”),结束

在目标硬件上运行可执行程序

使用以下命令从Linux运行可执行文件。
如果isunix系统('sshpass -p password SSH username@targetname "cd targetloc/arm_compute/;。/ logonet coderdemo_google.png”),结束
使用以下命令从Windows运行可执行文件。
如果ispc系统('plink.exe -l username -pw password targetname "cd targetloc/arm_compute/;。/ logonet coderdemo_google.png”),结束
5大预言 : ----------------------------- 谷歌99.992% 0.003%电晕0.003%啤酒0.001%埃索0.000%的联邦快递

另请参阅

||

相关的话题

Baidu
map