gx

chenyc

2025-06-12 7b72ac13a83764a662159d4a49b7fffb90476ecb

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
import { isValidProbablitiy } from '../utils/index';
import { IBoundingBox } from './BoundingBox';
import { LabeledBox } from './LabeledBox';
import { IRect } from './Rect';
 
export class PredictedBox extends LabeledBox {
  public static assertIsValidPredictedBox(box: any, callee: string) {
    LabeledBox.assertIsValidLabeledBox(box, callee);
 
    if (
      !isValidProbablitiy(box.score)
      || !isValidProbablitiy(box.classScore)
    ) {
      throw new Error(`${callee} - expected properties score (${box.score}) and (${box.classScore}) to be a number between [0, 1]`);
    }
  }
 
  private _score: number;
 
  private _classScore: number;
 
  constructor(box: IBoundingBox | IRect | any, label: number, score: number, classScore: number) {
    super(box, label);
    this._score = score;
    this._classScore = classScore;
  }
 
  public get score(): number { return this._score; }
 
  public get classScore(): number { return this._classScore; }
}