NSFW检测 (色情检测)

NSFW检测 (色情检测)

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

huggingface模型

项目github地址

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

使用方式一:python虚拟环境

  1. 安装 miniconda/anaconda
  2. clone项目
bash 复制代码
#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
  1. 构建虚拟环境
bash 复制代码
#构建虚拟环境
conda create -n nsfw_detection python=3.10
conda activate nsfw_detection

pip install torch
pip install transformers
pip install uvicorn
pip install gradio
  1. 运行服务
bash 复制代码
uvicorn --host 0.0.0.0 --port 7860 main:app

使用方式二:docker

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

bash 复制代码
# 镜像较大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"  #阿里云镜像

结语

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

相关推荐
唐某人丶几秒前
教你如何用 JS 实现 Agent 系统(2)—— 开发 ReAct 版本的“深度搜索”
前端·人工智能·aigc
FIT2CLOUD飞致云18 分钟前
九月月报丨MaxKB在不同规模医疗机构的应用进展汇报
人工智能·开源
阿里云大数据AI技术20 分钟前
【新模型速递】PAI-Model Gallery云上一键部署Qwen3-Next系列模型
人工智能
袁庭新39 分钟前
全球首位AI机器人部长,背负反腐重任
人工智能·aigc
机器之心1 小时前
谁说Scaling Law到头了?新研究:每一步的微小提升会带来指数级增长
人工智能·openai
算家计算1 小时前
AI配音革命!B站最新开源IndexTTS2本地部署教程:精准对口型,情感随心换
人工智能·开源·aigc
量子位1 小时前
马斯克周末血裁xAI 500人
人工智能·ai编程
算家计算1 小时前
OpenAI最强编程模型GPT-5-Codex发布!可独立编程7小时,编程效率提升10倍
人工智能·ai编程·资讯
聚客AI3 小时前
🌟大模型为什么产生幻觉?预训练到推理的漏洞全揭秘
人工智能·llm·掘金·日新计划
Juchecar3 小时前
一文讲清 nn.Sequential 等容器类
人工智能