26 lines
790 B
TypeScript
26 lines
790 B
TypeScript
|
import * as tf from '@tensorflow/tfjs-core';
|
||
|
|
||
|
import { disposeUnusedWeightTensors, ParamMapping } from '../common';
|
||
|
import { loadParamsFactory } from './loadParamsFactory';
|
||
|
import { TinyFaceFeatureExtractorParams } from './types';
|
||
|
|
||
|
export function extractParamsFromWeigthMapTiny(
|
||
|
weightMap: tf.NamedTensorMap
|
||
|
): { params: TinyFaceFeatureExtractorParams, paramMappings: ParamMapping[] } {
|
||
|
|
||
|
const paramMappings: ParamMapping[] = []
|
||
|
|
||
|
const {
|
||
|
extractDenseBlock3Params
|
||
|
} = loadParamsFactory(weightMap, paramMappings)
|
||
|
|
||
|
const params = {
|
||
|
dense0: extractDenseBlock3Params('dense0', true),
|
||
|
dense1: extractDenseBlock3Params('dense1'),
|
||
|
dense2: extractDenseBlock3Params('dense2')
|
||
|
}
|
||
|
|
||
|
disposeUnusedWeightTensors(weightMap, paramMappings)
|
||
|
|
||
|
return { params, paramMappings }
|
||
|
}
|