利用GoogleNet实现图片分类,能输出分类结果的那种,附代码
时间: 2023-06-24 19:03:11 浏览: 149
GoogleNet-Caffe模型实现图像分类.txt
以下是使用GoogleNet进行图像分类的Python代码示例:
```python
import tensorflow as tf
from tensorflow.keras.applications.inception_v3 import InceptionV3
from tensorflow.keras.preprocessing import image
from tensorflow.keras.applications.inception_v3 import preprocess_input, decode_predictions
import numpy as np
# 加载预训练模型
model = InceptionV3(weights='imagenet')
# 加载图像并进行预处理
img_path = 'example.jpg'
img = image.load_img(img_path, target_size=(299, 299))
x = image.img_to_array(img)
x = np.expand_dims(x, axis=0)
x = preprocess_input(x)
# 进行图像分类
preds = model.predict(x)
# 解码预测结果
print('Predicted:', decode_predictions(preds, top=3)[0])
```
注意,上述代码中的`example.jpg`是需要替换成你自己的图像路径。此外,你可以使用`top`参数来控制输出前几个预测结果。
阅读全文