27 lines
838 B
TypeScript
27 lines
838 B
TypeScript
import * as tf from '../../dist/tfjs.esm';
|
|
|
|
import { disposeUnusedWeightTensors, ParamMapping } from '../common/index';
|
|
import { loadParamsFactory } from './loadParamsFactory';
|
|
import { FaceFeatureExtractorParams } from './types';
|
|
|
|
export function extractParamsFromWeigthMap(
|
|
weightMap: tf.NamedTensorMap,
|
|
): { params: FaceFeatureExtractorParams, paramMappings: ParamMapping[] } {
|
|
const paramMappings: ParamMapping[] = [];
|
|
|
|
const {
|
|
extractDenseBlock4Params,
|
|
} = loadParamsFactory(weightMap, paramMappings);
|
|
|
|
const params = {
|
|
dense0: extractDenseBlock4Params('dense0', true),
|
|
dense1: extractDenseBlock4Params('dense1'),
|
|
dense2: extractDenseBlock4Params('dense2'),
|
|
dense3: extractDenseBlock4Params('dense3'),
|
|
};
|
|
|
|
disposeUnusedWeightTensors(weightMap, paramMappings);
|
|
|
|
return { params, paramMappings };
|
|
}
|