19 lines
No EOL
844 B
JavaScript
19 lines
No EOL
844 B
JavaScript
import { TfjsImageRecognitionBase } from 'tfjs-image-recognition-base';
|
|
export function extractParamsFromWeigthMap(weightMap) {
|
|
var paramMappings = [];
|
|
var extractWeightEntry = TfjsImageRecognitionBase.extractWeightEntryFactory(weightMap, paramMappings);
|
|
function extractFcParams(prefix) {
|
|
var weights = extractWeightEntry(prefix + "/weights", 2);
|
|
var bias = extractWeightEntry(prefix + "/bias", 1);
|
|
return { weights: weights, bias: bias };
|
|
}
|
|
var params = {
|
|
fc: {
|
|
age: extractFcParams('fc/age'),
|
|
gender: extractFcParams('fc/gender')
|
|
}
|
|
};
|
|
TfjsImageRecognitionBase.disposeUnusedWeightTensors(weightMap, paramMappings);
|
|
return { params: params, paramMappings: paramMappings };
|
|
}
|
|
//# sourceMappingURL=extractParamsFromWeigthMap.js.map
|