This library is meant to detect and censor nsfw content in public media. Models can be found here, they are downloaded automatically and cached in ~/.NsfwRecog.
NSFW Recog can either be used as stand-alone software or as module to be embedded in your python application.
- install requirements.txt
- install pytorch according to your cuda version
- (optional) install onnxruntime-gpu to support cuda usage
- Run one of the scripts listed in Usage
- pip install nsfw-recog
- Include the module as shown exemplarily in the demos listed in Usage
Instantiate a NsfwDetector class and use the methods detect(), blur(), video() and camera(). The demos show the usage:
- demo_detect.py: Detecting body parts/objects in an image
- demo_camera.py: Live detecting objects in a webcam stream
- demo_video.py: Detecting objects in a video and draw bounding boxes and labels to a new video
- demo_blur.py: Detecting objects in an image and blurring them.
I provide a basic model here. Please contact me if a better model with a higher accuracy is required.