/nsfw-classification

A Simple NSFW Classifier Based on Keras and Tensorflow

Primary LanguagePythonGNU Affero General Public License v3.0AGPL-3.0

NSFW Classification â›”

We Use Pretrained Keras Model to Classify NSFW Content

Requierments 📦

pip install -r requirements.txt

Classifier CLI Options 📋

--model     Model File Path                 #default: 'model/model.h5'
--dir       Directory Path To Classify      #default: 'data/frames'
--json      Json File Path                  #default: 'data/frames'

Then You Just Need To Run This

python main.py --dir PATH_TO_YOUR_DIR --model MODEL_FILE --jsob FOLDER_NAME

Frame Extractor CLI Options 📋

--frame     Frame Threshold     #default: 1800 (Every Minutes)
--src       Video File PATH     #default: 'sample.mp4'

Then You Just Need To Run This

python frame.py --frame FRAME_TH --src VIDEO_FILE

How To Serve Telegram Bot 🤖

Firstly, You Need To Get API_TOKEN from Bot_Fother and put it in .localenv File Then You Just Need To Run:

python bot.py

Rest API 📡

You Only Need To Run:

uvicorn server:app

Features ✨

  • Detect Neutral , Porn and Sexy
  • CLI
  • Telegram Bot
  • Video Files
  • Auto Scene Detect
  • Train Phase
  • Dataset
  • Rest API
    • Image Support
    • Video Support
    • GIF Support