NSFW检测 （色情检测）

NSFW检测 （色情检测）

最近因为工作需要对一些图片进行色情检测，在huggingface上找到一个nsfw检测模型，速度检测快，检测准确，还不用GPU，所以就在基础上将其封装成api，并用gradio编写了一个简单的测试界面

huggingface模型

项目github地址

api 测试详见项目api_test.py文件

使用方式一：python虚拟环境

1. 安装 miniconda/anaconda
2. clone项目

#clone项目
git clone https://github.com/a-zhui/nsfw_detection
#下载模型文件 模型文件较大（2.6G）
cd nsfw_detection #进入项目根目录
apt-get install git-lfs  #模型太大，以至于git仓库容纳不了。所以用git-lfs下载大文件
git lfs install
git clone https://huggingface.co/Falconsai/nsfw_image_detection

3. 构建虚拟环境

#构建虚拟环境
conda create -n nsfw_detection python=3.10
conda activate nsfw_detection

pip install torch
pip install transformers
pip install uvicorn
pip install gradio

4. 运行服务

uvicorn --host 0.0.0.0 --port 7860 main:app

使用方式二：docker

两个镜像地址镜像是相同的，考虑到dockerhub镜像拉取可能会慢，又整了个阿里云镜像（拉取dockerhub镜像时慢时快，玄学）

# 镜像较大12G
docker run -it -d -p 8080:7860 --restart always --workdir /root/nsfw_detection --name nsfw_detection bocai123/nsfw_detection:V1.0 bash -c "bash init.sh"
#docker run -it -d -p 8080:7860 --restart always --workdir /root/nsfw_detection --name nsfw_detection registry.cn-hangzhou.aliyuncs.com/bocai123/nsfw_detection:V1.0 bash -c "bash init.sh"  #阿里云镜像

结语

我也是闲的，写着玩，不过这个模型效果还是不错的