face-api/build/faceFeatureExtractor/extractParamsFromWeigthMapT...

14 lines
634 B
JavaScript
Raw Normal View History

2020-08-20 02:10:42 +02:00
import { disposeUnusedWeightTensors } from '../common';
import { loadParamsFactory } from './loadParamsFactory';
export function extractParamsFromWeigthMapTiny(weightMap) {
2020-08-18 14:04:33 +02:00
const paramMappings = [];
2020-08-20 02:10:42 +02:00
const { extractDenseBlock3Params } = loadParamsFactory(weightMap, paramMappings);
2020-08-18 14:04:33 +02:00
const params = {
dense0: extractDenseBlock3Params('dense0', true),
dense1: extractDenseBlock3Params('dense1'),
dense2: extractDenseBlock3Params('dense2')
};
2020-08-20 02:10:42 +02:00
disposeUnusedWeightTensors(weightMap, paramMappings);
2020-08-18 14:04:33 +02:00
return { params, paramMappings };
}
//# sourceMappingURL=extractParamsFromWeigthMapTiny.js.map