19 lines
874 B
JavaScript
19 lines
874 B
JavaScript
|
import { Box } from '../classes';
|
||
|
|
import { FaceDetection } from '../classes/FaceDetection';
|
||
|
|
import { isWithFaceDetection } from '../factories/WithFaceDetection';
|
||
|
|
import { round } from '../utils';
|
||
|
|
import { DrawBox } from './DrawBox';
|
||
|
|
export function drawDetections(canvasArg, detections) {
|
||
|
|
const detectionsArray = Array.isArray(detections) ? detections : [detections];
|
||
|
|
detectionsArray.forEach(det => {
|
||
|
|
const score = det instanceof FaceDetection
|
||
|
|
? det.score
|
||
|
|
: (isWithFaceDetection(det) ? det.detection.score : undefined);
|
||
|
|
const box = det instanceof FaceDetection
|
||
|
|
? det.box
|
||
|
|
: (isWithFaceDetection(det) ? det.detection.box : new Box(det));
|
||
|
|
const label = score ? `${round(score)}` : undefined;
|
||
|
|
new DrawBox(box, { label }).draw(canvasArg);
|
||
|
|
});
|
||
|
|
}
|
||
|
|
//# sourceMappingURL=drawDetections.js.map
|