本文属于机器翻译版本。若本译文内容与英语原文存在差异,则一律以英文原文为准。
教程:使用 TensorFlow Lite 对来自摄像机的图像进行示例图像分类推理
本教程向您展示如何使用 TensorFlow Lite 图像分类推理组件在 Greengrass 核心设备上在本地对来自摄像机的图像执行示例图像分类推理。此组件包括以下组件依赖关系:
-
TensorFlow Lite 图像分类模型存储组件
-
TensorFlow Lite 运行时组件
注意
本教程访问的是 Raspberry Pi
有关在 Greengrass 设备上进行机器学习的更多信息,请参阅执行机器学习推理。
先决条件
要完成本教程,您首先必须完成 教程:使用 TensorFlow Lite 执行示例图像分类推理。
您需要以下项目:
-
带有摄像机接口的 Linux Greengrass 核心设备。本教程访问以下受支持设备之一上的摄像机模块:
-
运行 Raspberry Pi OS
的 Raspberry Pi (以前称为 Raspbian)
有关设置 Greengrass 核心设备的信息,请参阅教程:入门 AWS IoT Greengrass V2。
核心设备必须满足以下要求:
-
在运行 Amazon Linux 2 或 Ubuntu 18.04 的 Greengrass 核心设备上,GNU C 库
(glibc)版本 2.27 或更高版本已安装在设备上。 -
在 Armv7l 设备(例如 Raspberry Pi)上,OpenCV-Python 的依赖关系已安装在设备上。运行以下命令以安装依赖关系。
sudo apt-get install libopenjp2-7 libilmbase23 libopenexr-dev libavcodec-dev libavformat-dev libswscale-dev libv4l-dev libgtk-3-0 libwebp-dev
-
运行 Raspberry Pi OS Bullseye 的 Raspberry Pi 设备必须满足以下要求:
-
NumPy 1.22.4 或更高版本已安装在设备上。Raspberry Pi OS Bullseye 包含早期版本的 NumPy,因此可以运行以下命令在设备上升级 NumPy。
pip3 install --upgrade numpy
-
旧摄像机堆栈已在设备上启用。Raspberry Pi OS Bullseye 包含一个新摄像机堆栈,该堆栈默认处于启用状态且不兼容,因此您必须启用旧摄像机堆栈。
启用旧摄像机堆栈
-
运行以下命令以打开 Raspberry Pi 配置工具。
sudo raspi-config
-
选择接口选项。
-
选择旧摄像机以启用旧摄像机堆栈。
-
重启 Raspberry Pi。
-
-
-
-
对于 Raspberry Pi 或 NVIDIA Jetson Nano 设备,Raspberry Pi 摄像机模块 V2 – 800 万像素,1080p
。要了解如何设置摄像机,请参阅 Raspberry Pi 文档中的连接摄像机 。
步骤 1:在设备上配置摄像机模块
在此步骤中,您将为设备安装并启用摄像机模块。在设备上运行以下命令。
步骤 2:验证您对默认通知主题的订阅
在 教程:使用 TensorFlow Lite 执行示例图像分类推理 中,您在 AWS IoT 控制台中将 AWS IoT MQTT 客户端配置为监视 TensorFlow Lite 图像分类组件就 ml/tflite/image-classification
主题发布的 MQTT 消息。在 AWS IoT 控制台中,验证此订阅是否存在。如果没有,请在将组件部署到 Greengrass 核心设备之前,按照 步骤 1:订阅默认通知主题 中的步骤订阅此主题。
步骤 3:修改 TensorFlow Lite 图像分类组件配置并进行部署
在此步骤中,您将配置 TensorFlow Lite 图像分类组件并将其部署到核心设备:
-
在 AWS IoT Greengrass 控制台
导航菜单中,选择组件。 -
在组件页面的公有组件选项卡上,选择
aws.greengrass.TensorFlowLiteImageClassification
。 -
在 aws.greengrass.TensorFlowLiteImageClassification 页面上,选择部署。
-
在添加到部署中,选择以下选项之一:
-
要将此组件合并到目标设备上的现有部署,请选择添加到现有部署,然后选择要修改的部署。
-
要在目标设备上创建新部署,请选择创建新部署。如果您的设备上已有部署,选择此步骤将替换现有部署。
-
-
在指定目标页面中,执行以下操作:
-
在部署信息下,输入或修改部署的友好名称。
-
在部署目标下,选择部署目标,然后选择下一步。如果您正在修改现有部署,则无法更改部署目标。
-
-
在选择组件页面的公有组件下,确认已选择
aws.greengrass.TensorFlowLiteImageClassification
组件,然后选择下一步。 -
在配置组件页面上,执行以下操作:
-
选择推理组件,然后选择配置组件。
-
在配置更新下的要合并的配置框中,输入以下配置更新。
{ "InferenceInterval": "60", "UseCamera": "true" }
通过此配置更新,该组件将访问您设备上的摄像机模块,并对摄像机拍摄的图像进行推理。推理代码每 60 秒运行一次。
-
选择确认,然后选择下一步。
-
-
在配置高级设置页面上,保留默认配置设置,然后选择下一步。
-
在检查页面上,选择部署。
-
创建一个
deployment.json
文件以定义 TensorFlow Lite 图像分类组件的部署配置。此文件应类似以下内容:{ "targetArn":"
targetArn
", "components": { "aws.greengrass.TensorFlowLiteImageClassification": { "componentVersion":2.1.0
, "configurationUpdate": { "InferenceInterval": "60", "UseCamera": "true" } } } }-
在
targetArn
字段中,按以下格式将
替换为部署目标的事物或事物组的 Amazon 资源名称(ARN):targetArn
-
事物:
arn:aws:iot:
region
:account-id
:thing/thingName
-
事物组:
arn:aws:iot:
region
:account-id
:thinggroup/thingGroupName
-
-
本教程使用组件版本 2.1.0。在
aws.greengrass.TensorFlowLiteImageClassification
组件对象中,将2.1.0
替换为使用不同版本的 TensorFlow Lite 图像分类组件。
通过此配置更新,该组件将访问您设备上的摄像机模块,并对摄像机拍摄的图像进行推理。推理代码每 60 秒运行一次。替换以下值
-
-
运行以下命令以在设备上部署 TensorFlow Lite 图像分类组件:
aws greengrassv2 create-deployment \ --cli-input-json file://
path/to/
deployment.json
完成部署可能需要数分钟。在下一步中,检查组件日志,以验证部署是否成功完成并查看推理结果。
步骤 4:查看推理结果
部署组件后,您可以在 Greengrass 核心设备的组件日志中和 AWS IoT 控制台的 AWS IoT MQTT 客户端中查看推理结果。要订阅组件所发布的推理结果的主题,请参阅步骤 2:验证您对默认通知主题的订阅。
主题
-
AWS IoT MQTT 客户端 – 要查看推理组件针对默认通知主题发布的结果,请完成以下步骤:
-
在 AWS IoT 控制台
导航菜单中,选择测试、MQTT 测试客户端。 -
在订阅下,选择
ml/tflite/image-classification
。
-
-
组件日志 – 要在组件日志中查看推理结果,请在 Greengrass 核心设备上运行以下命令。
sudo tail -f
/logs/aws.greengrass.TensorFlowLiteImageClassification.log/greengrass/v2
如果您在组件日志或 MQTT 客户端中看不到推理结果,则表示部署失败或未到达核心设备。如果您的核心设备未连接到互联网或没有运行组件所需的权限,则可能会发生这种情况。在核心设备上运行以下命令,以查看 AWS IoT Greengrass Core 软件日志文件。此文件包含来自 Greengrass 核心设备的部署服务的日志。
sudo tail -f
/logs/greengrass.log
/greengrass/v2
有关更多信息,请参阅 机器学习推理故障排除。
后续步骤
本教程向您展示如何使用 TensorFlow Lite 图像分类组件以及自定义配置选项对摄像机拍摄的图像执行示例图像分类。
有关自定义公有组件配置或创建自定义机器学习组件的更多信息,请参阅自定义您的机器学习组件。