双12视频智能分类购买

import cv2
import tensorflow as tf

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

cap = cv2.VideoCapture('example_video.mp4')

while cap.isOpened():
    ret, frame = cap.read()
    if not ret:
        break

    # 调整帧大小以适应模型输入
    image = cv2.resize(frame, (224, 224))
    image = tf.keras.applications.mobilenet_v2.preprocess_input(image)
    image = tf.expand_dims(image, 0)

    # 进行预测
    predictions = model.predict(image)
    decoded_predictions = tf.keras.applications.mobilenet_v2.decode_predictions(predictions, top=3)[0]

    for _, label, prob in decoded_predictions:
        print(f"{label}: {prob:.2f}")

    cv2.imshow('Video', frame)

    if cv2.waitKey(1) & 0xFF == ord('q'):
        break

cap.release()
cv2.destroyAllWindows()