dustturtle / nudity-coreml

A CoreML models that detects nudity in a picture

Nudity Detection for CoreML

TODO:需要比较下准确率（ph1ps的代码里面带坐标转换的输入，和我的nukedetector里面的转换）；个人感觉这里的输入不进行转换是更符合AI框架设计意图的.

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.