双十二内容识别选购

import tensorflow as tf
import numpy as np
from PIL import Image

# 加载预训练模型
model = tf.keras.applications.MobileNetV2(weights='imagenet')

def preprocess_image(image_path):
    img = Image.open(image_path)
    img = img.resize((224, 224))
    img_array = np.array(img) / 255.0
    img_array = np.expand_dims(img_array, axis=0)
    return img_array

def predict(image_path):
    img_array = preprocess_image(image_path)
    predictions = model.predict(img_array)
    decoded_predictions = tf.keras.applications.mobilenet_v2.decode_predictions(predictions, top=3)[0]
    return decoded_predictions

# 测试
image_path = 'path_to_your_image.jpg'
results = predict(image_path)
for result in results:
    print(f"{result[1]}: {result[2]*100:.2f}%")