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

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

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

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

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

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

04-14 05:42