pytorch移动端教程 Java调用libtorch

将模型转为libtorch模型之后
剩下的就是移动端是事情了
也就是使用java调用模型因为
libtorch 支持移动端且支持java调用
其实你使用java再电脑做做测试最好
但是不得不说libtorch 虽然支持java但是java教程却没有人看过没人用过
就从官方给的demo上学习一下
官方直接就给了个项目案例如下
首先克隆项目到电脑上
如果克隆慢请注册gitee
再git中fork这个项目
再使用gitee导入项目最后克隆即可
git clone https://github.com/pytorch/android-demo-app.git cd HelloWorldApp
这一步其实可以忽略，使用androidstudio 应该是自动配置的
./gradlew installDebug
重点就是改gradle dependencies

repositories {jcenter()
}dependencies {implementation 'org.pytorch:pytorch_android:1.4.0'implementation 'org.pytorch:pytorch_android_torchvision:1.4.0'
}

使用上面配置好的torch api 和 android的api读取图片
org.pytorch:pytorch_android_torchvision android.media.Image and android.graphics.Bitmap
在这个中org.pytorch.helloworld.MainActivity
使用这句话就能读取图片
Bitmap bitmap = BitmapFactory.decodeStream(getAssets().open("image.jpg"));
使用这句话Module module = Module.load(assetFilePath(this, "model.pt"));
就能加载模型并预处理归一化
Tensor inputTensor = TensorImageUtils.bitmapToFloat32Tensor(bitmap, TensorImageUtils.TORCHVISION_NORM_MEAN_RGB, TensorImageUtils.TORCHVISION_NORM_STD_RGB);
预测并得到结果输出的是1000的数组就是1000分类的概率
Tensor outputTensor = module.forward(IValue.from(inputTensor)).toTensor(); float[] scores = outputTensor.getDataAsFloatArray();
使用java得出那个概率最高(就是java中求数组最大值和最大值得索引)最后从类别名称列表中通过求出
最大值的索引得到这个图被识别为哪个类别
float maxScore = -Float.MAX_VALUE; int maxScoreIdx = -1; for (int i = 0; i < scores.length; i++) { if (scores[i] > maxScore) { maxScore = scores[i]; maxScoreIdx = i; } } String className = ImageNetClasses.IMAGENET_CLASSES[maxScoreIdx];
java 调用pytorch模型其实就是上面这样（重点在于前面的转化和后面的显示对于一个算法工程师来说后面是硬伤
因为这个是Android工程师做的事情，我们这都会了就是人工智能工程师全站，记得要求会移动端部署的工作一定要高工资因为
你做的是两个人的工作）
接下来就是调用摄像头并获取图的代码了
再这个文件org.pytorch.demo.vision.AbstractCameraXActivity的代码中使用下面代码即可
调用摄像头
`
void setupCameraX() {
final PreviewConfig previewConfig = new PreviewConfig.Builder().build();
final Preview preview = new Preview(previewConfig);
preview.setOnPreviewOutputUpdateListener(output -> mTextureView.setSurfaceTexture(output.getSurfaceTexture()));

final ImageAnalysisConfig imageAnalysisConfig =new ImageAnalysisConfig.Builder().setTargetResolution(new Size(224, 224)).setCallbackHandler(mBackgroundHandler).setImageReaderMode(ImageAnalysis.ImageReaderMode.ACQUIRE_LATEST_IMAGE).build();
final ImageAnalysis imageAnalysis = new ImageAnalysis(imageAnalysisConfig);
imageAnalysis.setAnalyzer((image, rotationDegrees) -> {analyzeImage(image, rotationDegrees);});CameraX.bindToLifecycle(this, preview, imageAnalysis);

}

void analyzeImage(android.media.Image, int rotationDegrees)`

实战部署过程可参考
https://blog.csdn.net/y_dd6011/article/details/104751029
到此为止理论已经大致了解了
接下来将会实战
多个简单的转换模型
打磨移动端部署的过程

而后弄一些复杂的网络进行转换
打磨转换模型这里

pytorch移动端教程 Java调用libtorch相关推荐

Jgit基础教程(Java调用git)
前言最近公司需要做一个java调用git的工具,这里简单的介绍了一下基本操作方法以及一些衍生的信息获取,或有不对的地方请大家批评指正.转载请注明出处. 一.Jgit依赖导入 <dependen ...
java调用云小蜜示例教程（调用sdk版本）
最近又用java写了云小蜜的调用教程,这里总结一下,首先maven依赖为: <!-- https://mvnrepository.com/artifact/com.aliyun/aliyun-j ...
java 调用笔记本摄像头拍照_教程：如何使用Java和C++在应用程序中实现面部识别
全文共8470字,预计学习时长25分钟来源:xaecong HOG:梯度方向直方图(histogram of orientedgradients)是一种图片描述符格式,它能够汇总图像(例如人脸)的主 ...
Java调用Pytorch实现以图搜图（附源码）
Java调用Pytorch实现以图搜图设计技术栈: 1.ElasticSearch环境: 2.Python运行环境(如果事先没有pytorch模型时,可以用python脚本创建模型): 1.运行效果 ...
Swig超详细入门教程（Java调用C/C++, CMake）——更新于2021.12
目录相关教程环境配置 0基础上手例子(C/C++) 使用CMake的例子(C语言) 使用CMake的例子(C++) 本文主要是手把手教萌新们如何用官方用例构建(有许多本人亲身踩坑血泪史) 相关教程 ...
android开发教程（3）— jni编程之采用SWIG从Java调用C/C++
2019独角兽企业重金招聘Python工程师标准>>> Android 从Java调用C/C++ ...
使用java调用C语言程序教程
1.idea创建一个java类严格来说,核心步骤并不是创建一个java类,而是创建一个方法,那个方法要被native修饰,这才是关键. 接着在TestNativeCode类当中声明我们的本地方法: ...
[pytorch] 官网教程+注释
pytorch官网教程+注释 Classifier import torch import torchvision import torchvision.transforms as transform ...
java xfire webservice 异步_[Java教程]Java中使用webservice，简化开发（xfire的webservice）...
[Java教程]Java中使用webservice,简化开发(xfire的webservice) 0 2016-01-07 15:00:11 首先,使用到的jar先导入项目中, xbean-sprin ...

pytorch移动端教程 Java调用libtorch

pytorch移动端教程 Java调用libtorch相关推荐

最新文章

热门文章