@vladmandic/face-api/src/ssdMobilenetv1/SsdMobilenetv1Options.ts

JavaScript module for Face Detection and Face Recognition Using Tensorflow/JS

export interface ISsdMobilenetv1Options {
  minConfidence?: number
  maxResults?: number
}

export class SsdMobilenetv1Options {
  protected _name: string = 'SsdMobilenetv1Options'

  private _minConfidence: number
  private _maxResults: number

  constructor({ minConfidence, maxResults }: ISsdMobilenetv1Options = {}) {
    this._minConfidence = minConfidence || 0.5
    this._maxResults = maxResults || 100

    if (typeof this._minConfidence !== 'number' || this._minConfidence <= 0 || this._minConfidence >= 1) {
      throw new Error(`${this._name} - expected minConfidence to be a number between 0 and 1`)
    }

    if (typeof this._maxResults !== 'number') {
      throw new Error(`${this._name} - expected maxResults to be a number`)
    }
  }

  get minConfidence(): number { return this._minConfidence }
  get maxResults(): number { return this._maxResults }
}