"use strict";
|
Object.defineProperty(exports, "__esModule", { value: true });
|
var common_1 = require("../common");
|
var loadConvParamsFactory_1 = require("../common/loadConvParamsFactory");
|
function loadParamsFactory(weightMap, paramMappings) {
|
var extractWeightEntry = common_1.extractWeightEntryFactory(weightMap, paramMappings);
|
var extractConvParams = loadConvParamsFactory_1.loadConvParamsFactory(extractWeightEntry);
|
var extractSeparableConvParams = common_1.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
|
