【亲测免费】 雅虎开放NSFW模型使用教程

1. 项目目录结构及介绍

雅虎的Open NSFW（Not Suitable for Work）项目主要目录结构如下：

• nsfw_model: 包含用于分类的深度学习模型文件。
• LICENSE.md: 项目授权文件，描述了项目的许可条款。
• README.md: 项目简介和使用指南。
• classify_nsfw.py: 主要的图像分类脚本，用于加载模型并进行NSFW评分。

这个项目旨在通过预先训练的Caffe模型检测不适宜工作的图像。

2. 项目启动文件介绍

classify_nsfw.py 是项目的核心启动文件，其功能包括：

• 加载预训练的Caffe模型权重文件。
• 处理输入图像，将其转换成模型可以处理的格式。
• 对输入图像进行NSFW分类，返回安全（SFW）分数和不安全（NSFW）分数。
• 支持两种不同的图像加载机制，可以通过命令行参数选择。

例如，您可以使用以下命令来运行这个脚本：

python classify_nsfw.py -m data/open_nsfw-weights.npy your_image.jpg

其中，-m 参数指定模型权重路径，your_image.jpg 是要检测的图像。

3. 项目的配置文件介绍

该项目并不直接使用传统意义上的配置文件。然而，classify_nsfw.py 接受一些命令行选项作为配置：

• -m 或 --model_weights: 指定模型权重文件的位置，这是运行分类所需的必要配置。
• -l 或 --image_loader: 决定如何加载图像，默认支持Yahoo和TensorFlow两种方式，但这里可能已过时，因为项目已经基于TensorFlow重构。
• -t 或 --input_type: 指定输入图像的类型，如原始二进制、Tensor或Base64编码的JPEG图像，但目前只支持JPEG。

使用这些参数可以根据具体需求定制程序的行为，比如使用自定义的模型权重文件或不同类型的输入图像数据。