图片审核秒杀

import tensorflow as tf
from tensorflow.keras.applications.mobilenet_v2 import preprocess_input
from tensorflow.keras.preprocessing import image
import numpy as np

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

def is_safe_image(img_path):
    img = image.load_img(img_path, target_size=(224, 224))
    x = image.img_to_array(img)
    x = np.expand_dims(x, axis=0)
    x = preprocess_input(x)

    preds = model.predict(x)
    decoded_preds = tf.keras.applications.mobilenet_v2.decode_predictions(preds, top=1)[0]

    # 假设我们只允许'安全'类别的图片
    safe_categories = ['cat', 'dog', 'flower']  # 示例安全类别
    if decoded_preds[0][1] in safe_categories:
        return True
    else:
        return False

# 使用示例
img_path = 'path_to_your_image.jpg'
if is_safe_image(img_path):
    print("图片审核通过")
else:
    print("图片审核未通过")