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

Initialize

ESM

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

CommonJS

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

Usage

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 |

Model

The model can be found here.

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