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

结语

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

相关推荐
JAVA学习通1 天前
JetLinks设备接入的认识与理解
运维·docker·容器·rocketmq
清风吹过1 天前
LSTM新架构论文分享6:LSTM+Transformer融合
论文阅读·人工智能·深度学习·神经网络·lstm·transformer
Hoking1 天前
LangChain4j集成SpringBoot接入百炼大模型(Qwen)
java·人工智能·spring boot·llm
jyan_敬言1 天前
【Docker】docker存储配置与管理
docker·容器·dubbo·学习方法
量子位1 天前
马斯克从英伟达挖人做AI游戏!第一步:研发世界模型
人工智能·ai编程
程序员大雄学编程1 天前
「机器学习笔记11」深入浅出:解密基于实例的学习(KNN算法核心原理与实践)
人工智能·笔记·机器学习
AI拉呱_1 天前
第145期《2025年AI现状报告》解读(三):安全篇
人工智能·安全
益企研究院1 天前
AICC2025 智算中心与算力服务论坛顺利举办
人工智能
千年人参1001 天前
AI 与神经网络:从理论到现代应用
人工智能·深度学习·神经网络
视***间1 天前
边缘计算的未来:从技术突围到规模化爆发的关键十年(智算未来,驱动创新——视程空间赋能智能时代核心AI算力)
人工智能·科技·边缘计算·ai算力·视程空间