Caffe2-iOS iOS实时推理演示框架
iOS 上的深度学习应用怎么跑?Caffe2-iOS就挺方便的。它把原来挺重的 Caffe2 搬到了手机上,支持你直接把训练好的模型丢进 iOS App 里跑推理,像图像分类、物体识别都不在话下。核心是用CoreML和Metal,所以性能也还不错,响应快,不卡顿。
摄像头实时演示也是它的一大亮点,你可以边拍照边让模型跑判断,用自己的模型和照片来测试,效果一目了然。想试 Tiny-YOLO?也可以加载,适合移动端,不太吃资源。
语言方面你不用担心,项目支持Swift和Objective-C,你用哪个都行。再加上OpenCV做预,比如图像裁剪、旋转,也方便调试模型效果。
如果你平时做 iOS 开发,又想试试 AI 模型的移动部署,这个项目值得一看。代码结构清晰,逻辑也比较直观,照着跑起来没啥门槛。如果你模型训练好,就能直接部署上线,适合用来做 AI Demo 或者快速原型测试。
如果你还不太熟 Caffe2,也可以先看看深入 Caffe:探索深度学习框架这篇文章,了解下框架原理。想试别的模型平台?你也可以看看PyTorch 的图像分类项目或者基于 TensorFlow 的 ResNet 实现,扩展下思路。
,Caffe2-iOS还挺适合做入门级移动 AI 开发的,既能实战也能练手。如果你手头有模型想在 iPhone 上跑跑看,这个项目适合拿来试水。
Caffe2-iOS:iOS实时演示中的Caffe2。 用您自己的模型和照片进行测试
预估大小:162个文件
.gitignore
1KB
libprotobuf.a
23.97MB
libprotobuf-lite.a
2.84MB
libCaffe2_CPU.a
77.11MB
libCAFFE2_NNPACK.a
1.13MB
Current
1B
core_c.h
77KB
types_c.h
57KB
imgproc_c.h
30KB
highgui_c.h
28KB
40.43MB
文件大小:
评论区