3.2.3

# 导入必要的库
import onnx
import numpy as np
from PIL import Image
import onnxruntime as ort

# 定义预处理函数，用于将图片转换为模型所需的输入格式
def preprocess(image_path):
    input_shape = (1, 1, 64, 64)    # 模型输入期望的形状，这里是 (N, C, H, W)，N=batch size, C=channels, H=height, W=width
    img = Image.open(image_path).convert('L')    # 打开图像文件并将其转换为灰度图  1分
    img = img.resize((64, 64), Image.ANTIALIAS)    # 调整图像大小到模型输入所需的尺寸
    img_data = np.array(img, dtype=np.float32)    # 将PIL图像对象转换为numpy数组，并确保数据类型是float32
    # 调整数组的形状以匹配模型输入的形状
    img_data = np.expand_dims(img_data, axis=0)  # 添加 batch 维度
    img_data = np.expand_dims(img_data, axis=1)  # 添加 channel 维度
    assert img_data.shape == input_shape, f"Expected shape {input_shape}, but got {img_data.shape}"    # 确保最终的形状与模型输入要求的形状一致
    return img_data    # 返回预处理后的图像数据

# 定义情感类别与数字标签的映射表 2分

emotion_table = {'____________':0, '____________':1, 'surprise':2, 'sadness':3, 'anger':4, 'disgust':5, 'fear':6, 'contempt':7}

---

emotion_table = {'neutral':0, 'happiness':1, 'surprise':2, 'sadness':3, 'anger':4, 'disgust':5, 'fear':6, 'contempt':7}
代码解释：

# 加载模型 2分

ort_session = ____________.____________('emotion-ferplus.onnx')    # 使用onnxruntime创建一个会话，用于加载并运行模型

---

ort_session = ort.InferenceSession('emotion-ferplus.onnx')    # 使用onnxruntime创建一个会话，用于加载并运行模型

# 加载本地图片并进行预处理 2分

input_data = ____________('____________')

---

input_data = preprocess('img_test.png')

# 准备输入数据，确保其符合模型输入的要求
ort_inputs = {ort_session.get_inputs()[0].name: input_data}    # ort_session.get_inputs()[0].name 是获取模型的第一个输入的名字

# 运行模型，进行预测 2分

ort_outs = ____________.____________(None, ort_inputs)

---

ort_outs = ort_session.run(None, ort_inputs)

代码解释：动词在后面，

# 解码模型输出，找到预测概率最高的情感类别 2分

predicted_label = ____________.____________(ort_outs[0])

---

predicted_label = np.argmax(ort_outs[0])

代码解释：概率最高argmax

# 根据预测的标签找到对应的情感名称 2分

predicted_emotion = list(____________.keys())[____________]

---

predicted_emotion = list(emotion_table.keys())[predicted_label]

列表

标签

# 输出预测的情感

print(f"Predicted emotion: {predicted_emotion}")