12345678910111213141516171819202122232425262728293031323334 |
- "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
|