Skip to content

Conversation

@Cryptoverso
Copy link

feat: add NSFW detection models (nsfw-detector and OpenNSFW2)

  • Added the nsfw_mobilenet2.224x224.h5 model to apps/pyserver/app/nsfw_model/, used for detecting NSFW (Not Safe for Work) content.
  • Integrated OpenNSFW2 model from UOL to enhance detection accuracy.
  • These models classify images into categories such as: porn, hentai, drawings, sexy, and neutral.

Example output:
{
"/tmp/tmp.jpg": {
"drawings": 0.0057,
"hentai": 0.0006,
"neutral": 0.3629,
"porn": 0.0116,
"sexy": 0.6192
}
}

  • Models align with the detection logic of https://nsfwjs.com.
  • Updated existing object detection endpoints under the /oracle route instead of the default path.

image

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant