iOS MachineLearning 系列(16)—— 几个常用的图片分类CoreML模型

本文涉及的产品
交互式建模 PAI-DSW,每月250计算时 3个月
模型训练 PAI-DLC,100CU*H 3个月
模型在线服务 PAI-EAS,A10/V100等 500元 1个月
简介: 对于图片识别分类的模型来说,其输入和输出都一样,输入都为图像参数,输入为两部分,一部分为最佳预测结果,一部分为可能得预测结果及其可信度。

iOS MachineLearning 系列(16)—— 几个常用的图片分类CoreML模型

在本系列的前面文章中,有介绍使用Vision框架中的VNClassifyImageRequest进行图片物体识别。Apple也推荐了几个常用的图像分类模型可供开发者使用。可以在如下地址直接下载:

https://developer.apple.com/machine-learning/models/

1 - 几个模型的简单介绍

对于图片识别分类的模型来说,其输入和输出都一样,输入都为图像参数,输入为两部分,一部分为最佳预测结果,一部分为可能得预测结果及其可信度。

Resnet50

Resnet50是一种神经网络分类模型,其大小为102.6M左右,其可以从1000种类别中检测输入图像中的主要元素,包括树木,动物,食物,车辆和人等。其测量的误差率为7.8%。

MobileNetV2

MobileNetV2是一种神经网络分类模型,其大小约为24.7M,其同样能从千余种类别中检测图像里的主要元素。其测量的准确率为74.7%。

SqueezeNet

SqueezeNet也是一种神经网络分类模型,其最大的特点是非常小巧,参数可以减少50倍。我们直接使用的模型大小仅有5M左右。

整体来说,Resnet50体积最大,其预测的精准度也相比会更好,但是对于图片比较清晰,内容比较单一的图像,这些模型都可以满足我们的正常需求。

2 - 使用示例

关于CoreML模型的使用,之前的文章有做详细的介绍,这里并没有额外需要注意的地方。我们直接将模型下载下来引入到Xcode工程中,Xcode会自动帮我们生成模型的使用代码,分别初始化3个模型的操作对象如下:

// mobileNet2模型
let mobileNet2 = try! MobileNetV2(configuration: MLModelConfiguration())
// resent50模型
let resnet50 = try! Resnet50(configuration: MLModelConfiguration())
// squeezeNet 模型
let squeezeNet = try! SqueezeNet(configuration: MLModelConfiguration())

对应的,创建输入参数:

let mobileNet2Input = try! MobileNetV2Input(imageWith: image.cgImage!)
let resnet50Input = try! Resnet50Input(imageWith: image.cgImage!)
let squeezeNetInput = try! SqueezeNetInput(imageWith: image.cgImage!)

进行预测:

// 进行图像识别
let mobileOutput = try! mobileNet2.prediction(input: mobileNet2Input)
let resnetOutput = try! resnet50.prediction(input: resnet50Input)
let squeezeOutput = try! squeezeNet.prediction(input: squeezeNetInput)

label.text = label.text?.appending("mobileNet2模型预测结果:\n最可能:\(mobileOutput.classLabel)\n\n")
label.text = label.text?.appending("resnet50模型预测结果:\n最可能:\(resnetOutput.classLabel)\n\n")
label.text = label.text?.appending("squeeze模型预测结果:\n最可能:\(squeezeOutput.classLabel)\n\n")

结果如下图所示:

可以看到,3种模型都正确的对图片进行了分类,正确的检测出了图片中的“猕猴”。

完整的示例代码可以在如下地址找到:

https://github.com/ZYHshao/MachineLearnDemo

目录
相关文章
|
8月前
|
JSON JavaScript 安全
iOS应用程序数据保护:如何保护iOS应用程序中的图片、资源和敏感数据
iOS应用程序数据保护:如何保护iOS应用程序中的图片、资源和敏感数据
74 1
|
iOS开发
iOS TextView插入表情或者图片后字体变大或变小
iOS TextView插入表情或者图片后字体变大或变小
122 1
|
存储 对象存储
10-iOS分类关联对象底层实现原理
10-iOS分类关联对象底层实现原理
113 0
|
Android开发 iOS开发
iOS 替换WebView网页图片为本地图片
iOS 替换WebView网页图片为本地图片
286 0
|
8月前
|
存储 缓存 安全
基于iOS平台的高效图片缓存策略实现
【4月更文挑战第22天】 在移动应用开发中,图片资源的加载与缓存是影响用户体验的重要因素之一。尤其对于iOS平台,由于设备存储空间的限制以及用户对流畅性的高要求,设计一种合理的图片缓存策略显得尤为关键。本文将探讨在iOS环境下,如何通过使用先进的图片缓存技术,包括内存缓存、磁盘缓存以及网络请求的优化,来提高应用的性能和响应速度。我们将重点分析多级缓存机制的设计与实现,并对可能出现的问题及其解决方案进行讨论。
|
8月前
|
存储 缓存 算法
实现iOS平台的高效图片缓存策略
【4月更文挑战第22天】在移动应用开发中,图片资源的处理是影响用户体验的重要因素之一。特别是对于图像资源密集型的iOS应用,如何有效地缓存图片以减少内存占用和提升加载速度,是开发者们面临的关键挑战。本文将探讨一种针对iOS平台的图片缓存策略,该策略通过结合内存缓存与磁盘缓存的机制,并采用先进的图片解码和异步加载技术,旨在实现快速加载的同时,保持应用的内存效率。
|
8月前
|
存储 缓存 编解码
实现iOS平台的高效图片缓存策略
【4月更文挑战第23天】在移动应用开发领域,尤其是图像处理密集型的iOS应用中,高效的图片缓存策略对于提升用户体验和节省系统资源至关重要。本文将探讨一种针对iOS平台设计的图片缓存方案,该方案通过结合内存缓存与磁盘缓存的多层次结构,旨在优化图片加载性能并降低内存占用。我们将深入分析其设计理念、核心组件以及在实际场景中的应用效果,同时对比其他常见缓存技术的优势与局限。
|
8月前
|
存储 Web App开发 Android开发
iOS不支持WebP格式图片解决方案和iPhone 7及其后硬件拍照的HEIC格式图片
iOS不支持WebP格式图片解决方案和iPhone 7及其后硬件拍照的HEIC格式图片
712 1
iOS不支持WebP格式图片解决方案和iPhone 7及其后硬件拍照的HEIC格式图片
|
8月前
按钮的image图片是非圆角,直接对UIButton设置圆角,iOS13系统没有圆角效果的问题及解决方案
按钮的image图片是非圆角,直接对UIButton设置圆角,iOS13系统没有圆角效果的问题及解决方案
57 0
|
8月前
|
iOS开发 网络架构 UED
ios app的分类与本质,感想
ios app的分类与本质,感想
84 0

热门文章

最新文章

下一篇
开通oss服务