移动端AI与ML应用开发：基于iOS和Android

在

iOS

中使用自定义模型

|

203

所以现在你可以逐字节地遍历缓冲区数据。每个第 4 个都是 alpha 通道部分，因此你可

以忽略它。否则，你可以读取该字节，将其值除以 255 以对其进行归一化，然后将归一

化后的值存储在当前索引处的

rgbBytes

中：

var index = 0

for component in bufferData.enumerated() {

let offset = component.offset

let isAlphaComponent = (offset % 4) == 3

guard !isAlphaComponent else { continue }

rgbBytes[index] = Float(component.element) / 255.0

index += 1

}

现在你有了归一化字节序列，对于解释器来说，它看起来像一个包含图像的张量，你可

以将其作为

Data

返回，如下所示：

return rgbBytes.withUnsafeBufferPointer(Data.init)

下一步是将这个

Data

对象传递给解释器并得到一个推理结果。

11.2.6

第

6

步：获取张量的推理

此时，我们已将图像中的数据格式化为

Float

，包含红色、绿色和蓝色通道的

Data

，其

中包含每个像素每个通道的归一化数据。当解释器读取此数据时，它会将其识别为输入

张量并逐个浮点地读取它。首先，让我们初始化解释器并为输入和输出张量分配内存。

你会在应用程序的

getLabelForData ...