loadParamsFactory.js 1.7 KB

12345678910111213141516171819202122232425262728293031323334
  1. "use strict";
  2. Object.defineProperty(exports, "__esModule", { value: true });
  3. var common_1 = require("../common");
  4. var loadConvParamsFactory_1 = require("../common/loadConvParamsFactory");
  5. function loadParamsFactory(weightMap, paramMappings) {
  6. var extractWeightEntry = common_1.extractWeightEntryFactory(weightMap, paramMappings);
  7. var extractConvParams = loadConvParamsFactory_1.loadConvParamsFactory(extractWeightEntry);
  8. var extractSeparableConvParams = common_1.loadSeparableConvParamsFactory(extractWeightEntry);
  9. function extractDenseBlock3Params(prefix, isFirstLayer) {
  10. if (isFirstLayer === void 0) { isFirstLayer = false; }
  11. var conv0 = isFirstLayer
  12. ? extractConvParams(prefix + "/conv0")
  13. : extractSeparableConvParams(prefix + "/conv0");
  14. var conv1 = extractSeparableConvParams(prefix + "/conv1");
  15. var conv2 = extractSeparableConvParams(prefix + "/conv2");
  16. return { conv0: conv0, conv1: conv1, conv2: conv2 };
  17. }
  18. function extractDenseBlock4Params(prefix, isFirstLayer) {
  19. if (isFirstLayer === void 0) { isFirstLayer = false; }
  20. var conv0 = isFirstLayer
  21. ? extractConvParams(prefix + "/conv0")
  22. : extractSeparableConvParams(prefix + "/conv0");
  23. var conv1 = extractSeparableConvParams(prefix + "/conv1");
  24. var conv2 = extractSeparableConvParams(prefix + "/conv2");
  25. var conv3 = extractSeparableConvParams(prefix + "/conv3");
  26. return { conv0: conv0, conv1: conv1, conv2: conv2, conv3: conv3 };
  27. }
  28. return {
  29. extractDenseBlock3Params: extractDenseBlock3Params,
  30. extractDenseBlock4Params: extractDenseBlock4Params
  31. };
  32. }
  33. exports.loadParamsFactory = loadParamsFactory;
  34. //# sourceMappingURL=loadParamsFactory.js.map