深度学习中使用已训练模型对某张图片进行预测的python代码
时间: 2024-02-03 17:03:36 浏览: 126
好的,以下是使用已训练的模型对图片进行预测的Python代码示例:
```python
import tensorflow as tf
import numpy as np
from PIL import Image
# 加载已训练模型
model = tf.keras.models.load_model('path/to/your/model')
# 加载图片并进行预处理
img = Image.open('path/to/your/image')
img = img.resize((224, 224)) # 模型输入尺寸为 224x224
img = np.array(img)
img = img / 255.0 # 归一化像素值
# 对图片进行预测
predictions = model.predict(np.array([img]))
predicted_class = np.argmax(predictions)
# 打印预测结果
print(f"Predicted class: {predicted_class}")
```
其中,`path/to/your/model` 是你已训练模型的存储路径,`path/to/your/image` 是你想要预测的图片的路径。你需要根据你的实际情况进行修改。
阅读全文