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"  #阿里云镜像

结语

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

相关推荐
golang学习记8 分钟前
VSCode Copilot 编码智能体实战指南:让 AI 自主开发,你只负责 Review!
人工智能
渡我白衣12 分钟前
深度学习进阶(八)——AI 操作系统的雏形:AgentOS、Devin 与多智能体协作
人工智能·深度学习
小醉你真好22 分钟前
17、Centos9 安装 1Panel
linux·docker·运维开发
万岳软件开发小城24 分钟前
AI数字人系统源码+AI数字人小程序开发:2025年热门AI项目
人工智能·开源·软件开发·app开发·ai数字人小程序·ai数字人系统源码
xiangzhihong833 分钟前
Spring Boot集成SSE实现AI对话的流式响应
人工智能·spring boot
羊羊小栈34 分钟前
基于知识图谱(Neo4j)和大语言模型(LLM)的图检索增强(GraphRAG)的台风灾害知识问答系统(vue+flask+AI算法)
人工智能·毕业设计·知识图谱·创业创新·neo4j·毕设·大作业
+wacyltd大模型备案算法备案38 分钟前
【大模型备案】全国有439个大模型通过生成式人工智能大模型备案!
人工智能
学不会就看44 分钟前
PyTorch 张量学习
人工智能·pytorch·学习
兰文彬44 分钟前
Pytorch环境安装指南与建议
人工智能·pytorch·python
CareyWYR1 小时前
每周AI论文速递(251013-251017)
人工智能