儿童美汤 - Python

简介

儿童美汤是一个使用Python编写的可以帮助父母过滤网络图片的开源工具。通过对输入的图片进行检测和分类，只保留合适的儿童友好图片，并剔除包含暴力、色情等不良内容的图片。

使用方法

1. 安装依赖库

   使用儿童美汤前需要安装以下依赖库：

   • opencv-python
   • keras
   • tensorflow

   可以使用pip命令安装依赖库：

   pip install opencv-python keras tensorflow
   

2. 从GitHub上下载并解压源代码

   git clone https://github.com/your_account/your_repository.git
   

3. 运行儿童美汤

   python child_beauty.py
   

   命令行参数说明：

   • -i 或 --input：输入图片的路径或文件名
   • -o 或 --output：输出图片的路径或文件名
   • -c 或 --confidence：图片分类的置信度阈值，默认为0.5

核心代码

以下是主要的几个函数的代码片段：

def load_image(img_path):
    """
    读入图片并转换为数组
    """
    img = cv2.imread(img_path)
    img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
    img_array = np.array(img)
    return img_array

def preprocess_image(img_array):
    """
    预处理图片
    """
    img = cv2.resize(img_array, (224, 224))
    img = img.astype("float") / 255.0
    img = np.expand_dims(img, axis=0)
    return img

def load_model(model_path):
    """
    载入训练好的模型
    """
    model = keras.models.load_model(model_path)
    return model

def predict_image(model, img_array, confidence_thresh=0.5):
    """
    对图片分类预测
    """
    prediction = model.predict(img_array)[0]
    if prediction[0] > confidence_thresh:
        return "violent"
    elif prediction[1] > confidence_thresh:
        return "porn"
    else:
        return "safe"

贡献者

• Alice
• Bob
• Charlie

开源协议

儿童美汤使用 MIT 开源协议。