图像分类

图像分类 连接器提供在 AWS IoT Greengrass Core 上运行的机器学习 (ML) 推理服务。这一本地推理服务使用由 Amazon SageMaker 图像分类算法训练的模型执行图像分类。

用户定义的 Lambda 函数使用 AWS IoT Greengrass机器学习 SDK 将推理请求提交给本地推理服务。该服务在本地运行推理，并返回输入映像属于特定类别的概率。

AWS IoT Greengrass 为多个平台提供图像分类连接器。

要求

这些连接器具有以下要求：

连接器参数

这些连接器提供以下参数。

创建连接器示例 (CLI)

以下 CLI 命令创建一个 ConnectorDefinition，它具有包含图像分类的初始版本。

在 AWS IoT Greengrass 控制台 中，您可以从该组的 Connectors (连接器)页面添加一个连接器。有关更多信息，请参阅 Greengrass 连接器入门（控制台）。

输入数据

这些连接器不接受 MQTT 消息作为输入数据。

输出数据

这些连接器不会将 MQTT 消息作为输出数据发布。

示例用法

下面的示例 Lambda 函数使用 AWS IoT Greengrass机器学习 SDK 与图像分类连接器进行交互。

该示例初始化一个开发工具包客户端，并同步调用该开发工具包的 invoke_inference_service 函数来调用本地推理服务。它会传入算法类型、服务名称、映像类型和映像内容。然后，该示例会解析服务响应以获取概率结果（预测）。

import logging
from threading import Timer

import numpy as np

import greengrass_machine_learning_sdk as ml

# We assume the inference input image is provided as a local file
# to this inference client Lambda function.
with open('/test_img/test.jpg', 'rb') as f:
    content = f.read()

client = ml.client('inference')

def infer():
    logging.info('invoking Greengrass ML Inference service')

    try:
        resp = client.invoke_inference_service(
            AlgoType='image-classification',
            ServiceName='imageClassification',
            ContentType='image/jpeg',
            Body=content
        )
    except ml.GreengrassInferenceException as e:
        logging.info('inference exception {}("{}")'.format(e.__class__.__name__, e))
        return
    except ml.GreengrassDependencyException as e:
        logging.info('dependency exception {}("{}")'.format(e.__class__.__name__, e))
        return

    logging.info('resp: {}'.format(resp))
    predictions = resp['Body'].read()
    logging.info('predictions: {}'.format(predictions))
    
    # The connector output is in the format: [0.3,0.1,0.04,...]
    # Remove the '[' and ']' at the beginning and end.
    predictions = predictions[1:-1]
    count = len(predictions.split(','))
    predictions_arr = np.fromstring(predictions, count=count, sep=',')

    # Perform business logic that relies on the predictions_arr, which is an array
    # of probabilities.
    
    # Schedule the infer() function to run again in one second.
    Timer(1, infer).start()
    return

infer()

def function_handler(event, context):
    return

AWS IoT Greengrass机器学习 SDK 中的 invoke_inference_service 函数接受以下参数：

在 AWS IoT Greengrass 核心上安装 MXNet 依赖项

要使用图像分类连接器，您必须在核心设备上安装 Apache MXNet 框架的依赖项。连接器使用该框架来处理 ML 模型。

AWS IoT Greengrass 提供了脚本来安装以下常见平台和设备的依赖项（或用作安装参考）。如果使用的是其他平台或设备，请参阅 MXNet 文档以了解您的配置。

安装 MXNet 依赖项之前，请确保设备上存在所需的系统库（具有指定的最低版本）。

日志记录和问题排查

根据您的组设置，事件和错误日志会写入到 CloudWatch Logs 和/或本地文件系统。此连接器中的日志使用前缀 LocalInferenceServiceName。如果连接器出现异常行为，请检查连接器日志。其中经常包含有用的调试信息，例如缺失 ML 库依赖项或连接器启动故障的原因。

如果将 AWS IoT Greengrass 组配置为写入本地日志，则连接器会将日志文件写入 greengrass-root/ggc/var/log/user/区域/aws/。 有关使用 Greengrass 日志记录的更多信息，请参阅利用 AWS IoT Greengrass 日志进行监控。

可以使用以下信息帮助解决图像分类连接器代理问题。

所需系统库

以下选项卡列出了每个图像分类连接器所需的系统库。

问题

sudo ln /dev/null /dev/raw1394

许可证

图像分类连接器包含以下第三方软件/许可：

此连接器按照 Greengrass 核心软件许可协议发布。

另请参阅

本页内容：

• 要求
• 参数
• 输入数据
• 输出数据
• 示例用法
• 安装 MXNet 依赖项
• 日志记录和问题排查
• 许可证
• 另请参阅