快速在安卓端验证深度学习算法模型

2024-06-17 15:48

本文主要是介绍快速在安卓端验证深度学习算法模型,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!

https://zhuanlan.zhihu.com/p/76909819

https://zhuanlan.zhihu.com/p/76909819

 

1、背景

​ 前段时间在知乎上溜达,看到 

糖心他爸

 大神的专栏-实战嵌入端的AI算法,进去一看,不得了,发现新大陆了,深度学习模型还能在安卓端这么玩的吗?

​ 一般对我们这种初级炼丹师,要验证算法在端上的能力以及实测效果。一般是这样的, 如图1(至于你们是不是这样我是不知道),中间验证过程又臭又长,中间有些环节还得看别人有没有空(只怪自己水平差)。

图1 验证流程

​ 那么看了大神的专栏,发现原来可以这么玩(嫌弃ing,说了半天还没告诉怎么玩)。

​ 以mnist分类为例,先贴下大神的部分代码,详细代码看大神github。

int main(void)
{std::string image_name = "./mnist_test.jpg";std::string model_name = "./mnist.mnn";int forward = MNN_FORWARD_CPU;// int forward = MNN_FORWARD_OPENCL;int precision  = 2;int power      = 0;int memory     = 0;int threads    = 1;int INPUT_SIZE = 28;cv::Mat raw_image    = cv::imread(image_name.c_str());int raw_image_height = raw_image.rows;int raw_image_width  = raw_image.cols; cv::Mat image;……auto net = std::shared_ptr<MNN::Interpreter>(MNN::Interpreter::createFromBuffer(modelBuffer, bufferSize));……auto session = net->createSession(config);………………// run networknet->runSession(session);…………// post processing stepsauto scores_dataPtr  = tensor_scores_host.host<float>();// softmaxfloat exp_sum = 0.0f;for (int i = 0; i < 10; ++i){float val = scores_dataPtr[i];exp_sum += val;}// get result idxint  idx = 0;float max_prob = -10.0f;for (int i = 0; i < 10; ++i){float val  = scores_dataPtr[i];float prob = val / exp_sum;if (prob > max_prob){max_prob = prob;idx      = i;}}printf("the result is %d\n", idx);return 0;
}

​ 看下来直接在android端就能用C++完成模型效果以及性能测试了,感觉就不用去求爷爷求爸爸了,开森啊(这里面的MNN是阿里的端上inference框架,感兴趣可以去GitHub上看看,MNN可以参考MNN文档,当然还有nihui大佬的NCNN ):

2、怎么玩

​ 开森归开森,但是怎么玩呢?把大神的专栏翻了一遍,没看对应教程,大佬可能觉得这个so easy,大家都会,哭!!!

​ 怎么办?看看大神代码好像有mk文件,查查mk文件干啥的呢?发现是安卓上类似makefile文件,那就查查如何用mk编译可执行文件吧,通过问问度娘以及谷歌爸爸,发现是可以实现的,那么大体方向是没错。

​ 那么怎么编译?怎么把对应的文件传到android上?怎么在android上执行呢?(怒问自己,垃圾,你咋这么多不会呢!!!)

​ 这个时候果断去抱安卓同学的大腿了!

​ 大腿开始教我怎么玩啦!!!

2.1 ndk编译

​ 这里先git clone 一下大神的github代码,前面opencv动态库,MNN动态库,mk文件大神都给你们弄好了,下下来改改或者直接用就行了。

​ 再下载android的NDK(我是mac电脑,如果你们是其它系统,行不行我不知道,我只管挖不管埋)。

​ 可以发现ndk下面有个ndk-build文件(安卓大腿说用着玩意编译)。

# 先设置环境
PATH=$PATH:<ndk-build所在的文件目录>
# 这里我们就以大神里面的mnist为例
cd MNN-APPLICATIONS/applications/mnist/jni
ndk-build
# 这里可以发现直接在mnist创建了libs目录
# 运行结果
[arm64-v8a] Install        : libMNN.so => libs/arm64-v8a/libMNN.so
[arm64-v8a] Install        : libMNN_CL.so => libs/arm64-v8a/libMNN_CL.so
[arm64-v8a] Install        : libMNN_Vulkan.so => libs/arm64-v8a/libMNN_Vulkan.so
[arm64-v8a] Install        : onnx_mnist => libs/arm64-v8a/onnx_mnist

​ 到这里基本编译结束了,那么怎么把它导到android端呢?

2.2 文件传输

​ 下面这个得用adb工具,至于这玩意怎么安装,是什么玩意,自己百度。

​ 搞了个rk3399,连上电脑。

# 先来条命令
adb devices#运行结果如下,可以看到我下面有个设备了,设备号CBI9SLBNWK
List of devices attached
CBI9SLBNWK  device# 切换一下root权限
adb root
# 运行结果
”adbd is already running as root“

​ 这样子我们待会进入adb shell就有root权限了,就可以为所欲为了。

正式push数据上去:

# 传输编译好的文件,我这边比较懒把整个目录都push上去
adb push MNN-APPLICATIONS /data

2.3 运行可执行文件

# 下面进入adb shell环境
adb shell
# 下面就跟我们在linux的操作差不多了
# 先查看我们上传的文件在不
ls /data
# 发现文件是在的
cd /data
cd /data/temp/MNN-APPLICATIONS/applications/mnist/onnx/libs/arm64-v8a
# 可以发现有个编译好的可执行文件 onnx_mnist
# 把 onnx_minist拷出来,拷到有图片的目录,这里可以不拷贝,只是跟cpp代码中模型文件跟测试文件相对路径保持一致就行
cp onnx_mnist ../../jni
cd ../../jni# 看下大神的onnx_mnist.cpp文件,发现在该目录下需要mnist_test.jpg以及mnist.mnn两个文件
# 把该目录test.jpg文件名改成对应名字
mv test.jpg mnist_test.jpg
# 把graph中,MNN转换的mnist.mnn拷出来
cp ./graphs/minist.mnn ./

​ 激动人心的时候到啦,终于可以运行了!(集齐七颗龙珠了,可以召唤神龙了)

./onnx_mnist
# 我去,出错啦
CANNOT LINK EXECUTABLE "./onnx_mnist": library "libMNN.so" not found

​ 难道是编译目录不对?折腾了半天,把动态库移来移去,移到当前木来,看了mk文件,发现在对应的位置也有对应的动态库。

OpenCV_BASE = ../../../../libraries/opencv-4.1
MNN_BASE    = ../../../../libraries/mnn

​ 搞了半天没办法了,加上 

糖心他爸

 大神微信,求大神指点。

​ 大神直接贴了一行代码:

export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:<对应动态库路径>

​ 我去,原来没有指定对应动态库路径,,补上:

export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:../libs/arm64-v8a

运行结果:

Invalide device for support vulkan
Invalide device for support vulkan
the result is 5

​ 感动,终于有运行结果了!

这篇关于快速在安卓端验证深度学习算法模型的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!



http://www.chinasem.cn/article/1069865

相关文章

C#集成DeepSeek模型实现AI私有化的流程步骤(本地部署与API调用教程)

《C#集成DeepSeek模型实现AI私有化的流程步骤(本地部署与API调用教程)》本文主要介绍了C#集成DeepSeek模型实现AI私有化的方法,包括搭建基础环境,如安装Ollama和下载DeepS... 目录前言搭建基础环境1、安装 Ollama2、下载 DeepSeek R1 模型客户端 ChatBo

SpringBoot快速接入OpenAI大模型的方法(JDK8)

《SpringBoot快速接入OpenAI大模型的方法(JDK8)》本文介绍了如何使用AI4J快速接入OpenAI大模型,并展示了如何实现流式与非流式的输出,以及对函数调用的使用,AI4J支持JDK8... 目录使用AI4J快速接入OpenAI大模型介绍AI4J-github快速使用创建SpringBoot

Python与DeepSeek的深度融合实战

《Python与DeepSeek的深度融合实战》Python作为最受欢迎的编程语言之一,以其简洁易读的语法、丰富的库和广泛的应用场景,成为了无数开发者的首选,而DeepSeek,作为人工智能领域的新星... 目录一、python与DeepSeek的结合优势二、模型训练1. 数据准备2. 模型架构与参数设置3

golang字符串匹配算法解读

《golang字符串匹配算法解读》文章介绍了字符串匹配算法的原理,特别是Knuth-Morris-Pratt(KMP)算法,该算法通过构建模式串的前缀表来减少匹配时的不必要的字符比较,从而提高效率,在... 目录简介KMP实现代码总结简介字符串匹配算法主要用于在一个较长的文本串中查找一个较短的字符串(称为

Python爬虫selenium验证之中文识别点选+图片验证码案例(最新推荐)

《Python爬虫selenium验证之中文识别点选+图片验证码案例(最新推荐)》本文介绍了如何使用Python和Selenium结合ddddocr库实现图片验证码的识别和点击功能,感兴趣的朋友一起看... 目录1.获取图片2.目标识别3.背景坐标识别3.1 ddddocr3.2 打码平台4.坐标点击5.图

通俗易懂的Java常见限流算法具体实现

《通俗易懂的Java常见限流算法具体实现》:本文主要介绍Java常见限流算法具体实现的相关资料,包括漏桶算法、令牌桶算法、Nginx限流和Redis+Lua限流的实现原理和具体步骤,并比较了它们的... 目录一、漏桶算法1.漏桶算法的思想和原理2.具体实现二、令牌桶算法1.令牌桶算法流程:2.具体实现2.1

使用Python快速实现链接转word文档

《使用Python快速实现链接转word文档》这篇文章主要为大家详细介绍了如何使用Python快速实现链接转word文档功能,文中的示例代码讲解详细,感兴趣的小伙伴可以跟随小编一起学习一下... 演示代码展示from newspaper import Articlefrom docx import

0基础租个硬件玩deepseek,蓝耘元生代智算云|本地部署DeepSeek R1模型的操作流程

《0基础租个硬件玩deepseek,蓝耘元生代智算云|本地部署DeepSeekR1模型的操作流程》DeepSeekR1模型凭借其强大的自然语言处理能力,在未来具有广阔的应用前景,有望在多个领域发... 目录0基础租个硬件玩deepseek,蓝耘元生代智算云|本地部署DeepSeek R1模型,3步搞定一个应

Deepseek R1模型本地化部署+API接口调用详细教程(释放AI生产力)

《DeepseekR1模型本地化部署+API接口调用详细教程(释放AI生产力)》本文介绍了本地部署DeepSeekR1模型和通过API调用将其集成到VSCode中的过程,作者详细步骤展示了如何下载和... 目录前言一、deepseek R1模型与chatGPT o1系列模型对比二、本地部署步骤1.安装oll

Java深度学习库DJL实现Python的NumPy方式

《Java深度学习库DJL实现Python的NumPy方式》本文介绍了DJL库的背景和基本功能,包括NDArray的创建、数学运算、数据获取和设置等,同时,还展示了如何使用NDArray进行数据预处理... 目录1 NDArray 的背景介绍1.1 架构2 JavaDJL使用2.1 安装DJL2.2 基本操