suko nsfw - veka-server/onnx-php GitHub Wiki

suko / nsfw

model sources : https://huggingface.co/suko/nsfw

require_once(__DIR__.'/../vendor/autoload.php');

Onnx\Library::setFolder(__DIR__.'/../');
Onnx\Library::install();

$ia = new Onnx\Task\Vision(config:[
    'tags' => [ 0 => "Naked", 1 => "Safe"]
    ,'rescale_factor' => 0.00392156862745098
    ,'format' => 'rgb'
    ,'height' => 224
    ,'width' => 224
    ,'shape' => 'bhwc'  /* batch channel height width */
    ,'modelNameOrPath' => __DIR__.'/../models/model_suko_nsfw.onnx'
]);

$ia->loadModel();
$tags = $ia->getTags( __DIR__ . '\images\sexy.jpg');
var_dump($tags);