UNPKG

@timebutt/face-api.js

Version:

JavaScript API for face detection and face recognition in the browser with tensorflow.js

19 lines 807 B
import { disposeUnusedWeightTensors, extractWeightEntryFactory } from '../common'; export function extractParamsFromWeigthMap(weightMap) { var paramMappings = []; var extractWeightEntry = extractWeightEntryFactory(weightMap, paramMappings); function extractFcParams(prefix) { var weights = extractWeightEntry("".concat(prefix, "/weights"), 2); var bias = extractWeightEntry("".concat(prefix, "/bias"), 1); return { weights: weights, bias: bias }; } var params = { fc: { age: extractFcParams('fc/age'), gender: extractFcParams('fc/gender') } }; disposeUnusedWeightTensors(weightMap, paramMappings); return { params: params, paramMappings: paramMappings }; } //# sourceMappingURL=extractParamsFromWeigthMap.js.map