一个python包,用于使用caffe和yahoo的open-nsfw模型分析图像。
nsfw的Python项目详细描述
python nsfw
一个python包,用于使用Caffe和Yahoo's open_nsfw模型分析图像。
先决条件
python nsfw需要python 3并在caffe兼容系统上运行。
咖啡
如果您使用的是MacOS和Homebrew,安装Caffe的最简单方法是使用自制程序:
brew install caffe
或者,您可以使用Dockerlinux映像(例如^{
apt update && apt install caffe-cpu python3 python3-pip wget
安装
python nsfw可以使用pip:
pip3installnsfw
用法
importPIL.ImageasImagefromnsfwimportclassifyimage=Image.open("/path/to/image.jpg")sfw,nsfw=classify(image)print("SFW Probability: {}".format(sfw))print("NSFW Probability: {}".format(nsfw))
许可证
python nsfw在MIT License下免费提供。用它,从中学习,用它,改进它,改变它,根据你的需要来调整它。
python nsfw附带了yahoo的open_nsfw Caffe models(nsfw/deploy.prototxt
和nsfw/resent_50_1by2_nsfw.caffemodel
)的副本,这些副本是根据BSD 2-Clause License授权的。