OpenVINO（compiled_model.input()）

• 1. 函数概述
• 2. 使用场景
• 3. 返回的对象属性
• 4. 示例代码
• 5. 多输入情况
• 6. 常见用途
• 7. 注意事项

compiled_model.input() 是 OpenVINO API 中的一个方法，用于获取编译后的模型输入节点的信息。

1. 函数概述

• compiled_model.input() 返回的是一个对象或列表，包含编译后模型的输入节点信息。主要用于在推理过程中获取输入的详细属性，例如输入节点的名称、数据类型、维度等。
• 这个方法通常在模型加载和编译完成后调用，以便在推理前检查或设置输入的格式。

2. 使用场景

在使用 OpenVINO 进行推理时，需要知道模型的输入节点信息，以确保传入的数据与模型的输入规格匹配。compiled_model.input() 提供了对这些信息的访问，便于用户动态适配输入数据。

3. 返回的对象属性

返回的对象通常具备以下属性：

• Name（名称）：输入节点的名称，便于在推理时通过名称指定输入。
• Shape（形状）：输入数据的维度，例如 [1, 3, 224, 224] 表示一个批次大小为 1、通道数为 3、分辨率为 224x224 的图像输入。
• Data Type（数据类型）：输入数据的类型，例如 float32 或 uint8，用于确保输入数据类型的兼容性。

4. 示例代码

import openvino.runtime as ov

# 加载模型和编译
core = ov.Core()
model = core.read_model("model.xml")
compiled_model = core.compile_model(model, "CPU")

# 获取输入节点信息
input_info = compiled_model.input()

# 打印输入信息
print("Input name:", input_info.get_any_name())
print("Input shape:", input_info.shape)
print("Input data type:", input_info.get_element_type())

5. 多输入情况

对于多输入模型，compiled_model.input() 返回一个输入节点的列表，每个输入节点对应模型的一个输入。可以遍历每个输入节点的属性：

for input_node in compiled_model.inputs:
    print("Input name:", input_node.get_any_name())
    print("Input shape:", input_node.shape)
    print("Input data type:", input_node.get_element_type())

6. 常见用途

• 确认输入格式：在实际推理前检查模型输入的形状、数据类型，以确保输入数据符合模型的要求。
• 输入数据预处理：根据输入节点的信息对数据进行适当的预处理，比如调整图像尺寸、格式、归一化等。

7. 注意事项

• 若模型输入的数据格式不符合预期，推理会报错，因此建议在运行推理前始终检查输入节点的属性。
• 不同模型的输入可能有多个节点和不同数据格式，务必根据实际模型配置输入。

通过 compiled_model.input()，可以有效获取模型输入的详细信息，使推理前的数据准备更加精准。