23 lines
1.1 KiB
JavaScript
23 lines
1.1 KiB
JavaScript
|
export var TinyYolov2SizeType;
|
||
|
|
(function (TinyYolov2SizeType) {
|
||
|
|
TinyYolov2SizeType[TinyYolov2SizeType["XS"] = 224] = "XS";
|
||
|
|
TinyYolov2SizeType[TinyYolov2SizeType["SM"] = 320] = "SM";
|
||
|
|
TinyYolov2SizeType[TinyYolov2SizeType["MD"] = 416] = "MD";
|
||
|
|
TinyYolov2SizeType[TinyYolov2SizeType["LG"] = 608] = "LG";
|
||
|
|
})(TinyYolov2SizeType || (TinyYolov2SizeType = {}));
|
||
|
|
export class TinyYolov2Options {
|
||
|
|
constructor({ inputSize, scoreThreshold } = {}) {
|
||
|
|
this._name = 'TinyYolov2Options';
|
||
|
|
this._inputSize = inputSize || 416;
|
||
|
|
this._scoreThreshold = scoreThreshold || 0.5;
|
||
|
|
if (typeof this._inputSize !== 'number' || this._inputSize % 32 !== 0) {
|
||
|
|
throw new Error(`${this._name} - expected inputSize to be a number divisible by 32`);
|
||
|
|
}
|
||
|
|
if (typeof this._scoreThreshold !== 'number' || this._scoreThreshold <= 0 || this._scoreThreshold >= 1) {
|
||
|
|
throw new Error(`${this._name} - expected scoreThreshold to be a number between 0 and 1`);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
get inputSize() { return this._inputSize; }
|
||
|
|
get scoreThreshold() { return this._scoreThreshold; }
|
||
|
|
}
|
||
|
|
//# sourceMappingURL=TinyYolov2Options.js.map
|