face-api/src/ssdMobilenetv1/SsdMobilenetv1Options.ts

30 lines
876 B
TypeScript

export interface ISsdMobilenetv1Options {
minConfidence?: number
maxResults?: number
}
export class SsdMobilenetv1Options {
protected _name = '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; }
}