Android端高性能图像分类解决方案

时间 2019-12-10

标签 android 高性能图像分类解决方案栏目 Android 繁體版

原文原文链接

因为公司业务须要,前段时间开始了解AI方面的东西,准备找一个在android端性能较高的前向计算框架,了解了tflite,百度的mdl和腾讯的ncnn,最终敲定ncnn,不失所望,效果很不错,基本达到预期目标,大致流程以下:android

1.选择适用的网络结构,我选的是shufflenet(由于是移动端,性能为上)网络

2.使用caffe训练相应网络对应的模型(炼丹,此步骤坑比较多,对于AI门外汉来讲只能作一些调参,基本动不了网络结构)框架

3.将caffe训练好的模型转换为ncnn支持的模型(ncnn有相应转换工具和教程)工具

4.编写jni,调用转换好的模型进行预测性能

不得不说,ncnn真不愧为众多开源框架中的最良心的了,教程完备,更新快,速度快教程