Caffe2-iOS iOS实时推理演示框架

iOS 上的深度学习应用怎么跑?Caffe2-iOS就挺方便的。它把原来挺重的 Caffe2 搬到了手机上,支持你直接把训练好的模型丢进 iOS App 里跑推理,像图像分类、物体识别都不在话下。核心是用CoreMLMetal,所以性能也还不错,响应快,不卡顿。

摄像头实时演示也是它的一大亮点,你可以边拍照边让模型跑判断,用自己的模型和照片来测试,效果一目了然。想试 Tiny-YOLO?也可以加载,适合移动端,不太吃资源。

语言方面你不用担心,项目支持SwiftObjective-C,你用哪个都行。再加上OpenCV做预,比如图像裁剪、旋转,也方便调试模型效果。

如果你平时做 iOS 开发,又想试试 AI 模型的移动部署,这个项目值得一看。代码结构清晰,逻辑也比较直观,照着跑起来没啥门槛。如果你模型训练好,就能直接部署上线,适合用来做 AI Demo 或者快速原型测试。

如果你还不太熟 Caffe2,也可以先看看深入 Caffe:探索深度学习框架这篇文章,了解下框架原理。想试别的模型平台?你也可以看看PyTorch 的图像分类项目或者基于 TensorFlow 的 ResNet 实现,扩展下思路。

Caffe2-iOS还挺适合做入门级移动 AI 开发的,既能实战也能练手。如果你手头有模型想在 iPhone 上跑跑看,这个项目适合拿来试水。

folder
Caffe2-iOS:iOS实时演示中的Caffe2。 用您自己的模型和照片进行测试 预估大小:162个文件
file
.gitignore 1KB
file
libprotobuf.a 23.97MB
file
libprotobuf-lite.a 2.84MB
file
libCaffe2_CPU.a 77.11MB
file
libCAFFE2_NNPACK.a 1.13MB
file
Current 1B
file
core_c.h 77KB
file
types_c.h 57KB
file
imgproc_c.h 30KB
file
highgui_c.h 28KB
file
kmeans_index.h 35KB
file
lsh_table.h 18KB
file
autotuned_index.h 21KB
file
kdtree_index.h 19KB
file
dist.h 28KB
file
lsh_index.h 16KB
file
kdtree_single_index.h 20KB
file
hierarchical_clustering_index.h 22KB
file
libCAFFE2_PTHREADPOOL.a 41KB
file
.gitattributes 166B
zip 文件大小:40.43MB