@timebutt/face-api.js
Version:
JavaScript API for face detection and face recognition in the browser with tensorflow.js
10 lines • 647 B
JavaScript
import * as tf from '@tensorflow/tfjs-core';
export function extractConvParamsFactory(extractWeights, paramMappings) {
return function (channelsIn, channelsOut, filterSize, mappedPrefix) {
var filters = tf.tensor4d(extractWeights(channelsIn * channelsOut * filterSize * filterSize), [filterSize, filterSize, channelsIn, channelsOut]);
var bias = tf.tensor1d(extractWeights(channelsOut));
paramMappings.push({ paramPath: "".concat(mappedPrefix, "/filters") }, { paramPath: "".concat(mappedPrefix, "/bias") });
return { filters: filters, bias: bias };
};
}
//# sourceMappingURL=extractConvParamsFactory.js.map