34 lines
No EOL
1.9 KiB
JavaScript
34 lines
No EOL
1.9 KiB
JavaScript
"use strict";
|
|
Object.defineProperty(exports, "__esModule", { value: true });
|
|
var tfjs_image_recognition_base_1 = require("tfjs-image-recognition-base");
|
|
var loadConvParamsFactory_1 = require("../common/loadConvParamsFactory");
|
|
function loadParamsFactory(weightMap, paramMappings) {
|
|
var extractWeightEntry = tfjs_image_recognition_base_1.TfjsImageRecognitionBase.extractWeightEntryFactory(weightMap, paramMappings);
|
|
var extractConvParams = loadConvParamsFactory_1.loadConvParamsFactory(extractWeightEntry);
|
|
var extractSeparableConvParams = tfjs_image_recognition_base_1.TfjsImageRecognitionBase.loadSeparableConvParamsFactory(extractWeightEntry);
|
|
function extractDenseBlock3Params(prefix, isFirstLayer) {
|
|
if (isFirstLayer === void 0) { isFirstLayer = false; }
|
|
var conv0 = isFirstLayer
|
|
? extractConvParams(prefix + "/conv0")
|
|
: extractSeparableConvParams(prefix + "/conv0");
|
|
var conv1 = extractSeparableConvParams(prefix + "/conv1");
|
|
var conv2 = extractSeparableConvParams(prefix + "/conv2");
|
|
return { conv0: conv0, conv1: conv1, conv2: conv2 };
|
|
}
|
|
function extractDenseBlock4Params(prefix, isFirstLayer) {
|
|
if (isFirstLayer === void 0) { isFirstLayer = false; }
|
|
var conv0 = isFirstLayer
|
|
? extractConvParams(prefix + "/conv0")
|
|
: extractSeparableConvParams(prefix + "/conv0");
|
|
var conv1 = extractSeparableConvParams(prefix + "/conv1");
|
|
var conv2 = extractSeparableConvParams(prefix + "/conv2");
|
|
var conv3 = extractSeparableConvParams(prefix + "/conv3");
|
|
return { conv0: conv0, conv1: conv1, conv2: conv2, conv3: conv3 };
|
|
}
|
|
return {
|
|
extractDenseBlock3Params: extractDenseBlock3Params,
|
|
extractDenseBlock4Params: extractDenseBlock4Params
|
|
};
|
|
}
|
|
exports.loadParamsFactory = loadParamsFactory;
|
|
//# sourceMappingURL=loadParamsFactory.js.map
|