11.11图片内容安全智能审核推荐

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=3)[0]

    # 根据预测结果判断是否安全
    for pred in decoded_preds:
        if pred[1] in ['nudity', 'violence']:  # 假设这些是不安全类别
            return False
    return True

# 使用示例
img_path = 'path_to_your_image.jpg'
print(is_safe_image(img_path))