This library detects nsfw objects and body parts in images and videos and optionally censores them.
Project description
NSFW Recog
This library is meant to detect and censor nsfw content in public media. Models can be found here, they are downloaded automatically and cached.
Installation
- install requirements.txt
- install pytorch according to your cuda version
- (optional) install onnxruntime-gpu to support cuda usage
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.
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
nsfwrecog-1.0.tar.gz
(8.4 kB
view hashes)