• This repository has been archived on 06/Nov/2021
  • Stars
    star
    103
  • Rank 333,046 (Top 7 %)
  • Language
    Swift
  • License
    MIT License
  • Created about 7 years ago
  • Updated almost 7 years ago

Reviews

There are no reviews yet. Be the first to send feedback to the community and the maintainers!

Repository Details

A CoreML models that detects nudity in a picture

Nudity Detection for CoreML

Description

This is the OpenNSFW dataset implemented in Apple's new framework called CoreML. The OpenNSFW dataset can predict images as either SFW (safe for work) or NSFW (not safe for work) from images. The model was built with Caffe and is a fine-tuned Resnet model.

To test this model you can open the Nudity.xcodeproj and run it on your device (iOS 11 and Xcode 9 is required). To test further images just add them to the project and replace my testing with yours.

Obtaining the model

  • Download the model from Google Drive and drag it right into your project folder
  • Convert the model on your own:
    1. Change directory: cd ./Convert
    2. Change directory: sh convert.sh

More information

If you want to find out more how this model works and on which data it was trained on, feel free to visit the original OpenNSFW page on Github

Examples

I won't show any NSFW examples for obvious reasons.