face-api/build/faceExpressionNet/FaceExpressions.js

21 lines
937 B
JavaScript
Raw Normal View History

"use strict";
Object.defineProperty(exports, "__esModule", { value: true });
exports.FaceExpressions = exports.FACE_EXPRESSION_LABELS = void 0;
exports.FACE_EXPRESSION_LABELS = ['neutral', 'happy', 'sad', 'angry', 'fearful', 'disgusted', 'surprised'];
class FaceExpressions {
2020-08-18 14:04:33 +02:00
constructor(probabilities) {
if (probabilities.length !== 7) {
throw new Error(`FaceExpressions.constructor - expected probabilities.length to be 7, have: ${probabilities.length}`);
}
exports.FACE_EXPRESSION_LABELS.forEach((expression, idx) => {
2020-08-18 14:04:33 +02:00
this[expression] = probabilities[idx];
});
}
asSortedArray() {
return exports.FACE_EXPRESSION_LABELS
2020-08-18 14:04:33 +02:00
.map(expression => ({ expression, probability: this[expression] }))
.sort((e0, e1) => e1.probability - e0.probability);
}
}
exports.FaceExpressions = FaceExpressions;
2020-08-18 14:04:33 +02:00
//# sourceMappingURL=FaceExpressions.js.map