如何将加载的keras模型的图像保存为png/jpg？

import numpy as np
from PIL import Image
import tensorflow as tf
from tensorflow.keras.models import load_model

# 加载Keras模型
model = load_model('path_to_your_model.h5')

# 假设你有一个图像数据img_data，形状为(1, height, width, channels)
# 这里我们生成一个随机图像作为示例
img_data = np.random.rand(1, 224, 224, 3)

# 进行预测
predictions = model.predict(img_data)

# 假设我们只关心第一个预测结果
predicted_class = np.argmax(predictions, axis=1)[0]

# 将图像数据转换为PIL图像对象
img = Image.fromarray((img_data[0] * 255).astype(np.uint8))

# 保存图像为PNG格式
img.save('predicted_image.png')

print(f'Predicted class: {predicted_class}')
print('Image saved as predicted_image.png')