private-detector-js

This package provides the necessary image preprocessing to use Bumble's Private Detector™ model in JavaScript or TypeScript.

Install

npm install private-detector-js

const { RunInference } = import 'private-detector-js'

const { RunInference } = require('private-detector-js')

const probabilities = await RunInference(modelPath, filePaths, options)

Parameter	Type	Description
modelPath	String	path to the model.json file
filePaths	Array	path(s) to the image(s) to be evaulated
options	Object	options to manually provide paths to the model's weight files. Currently, weightUrlConverter and weightPathPrefix are supported. More information here

The model can be found here.

This model was converted from Python using TensorFlow.js's converter tool, as described here.