nudenet‐classifier - veka-server/onnx-php GitHub Wiki

nudenet-classifier

model sources : https://github.com/notAI-tech/NudeNet

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

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

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

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