loadParamsFactory.js 1.6 KB

12345678910111213141516171819202122232425262728293031
  1. import { extractWeightEntryFactory, loadSeparableConvParamsFactory } from '../common';
  2. import { loadConvParamsFactory } from '../common/loadConvParamsFactory';
  3. export function loadParamsFactory(weightMap, paramMappings) {
  4. var extractWeightEntry = extractWeightEntryFactory(weightMap, paramMappings);
  5. var extractConvParams = loadConvParamsFactory(extractWeightEntry);
  6. var extractSeparableConvParams = loadSeparableConvParamsFactory(extractWeightEntry);
  7. function extractDenseBlock3Params(prefix, isFirstLayer) {
  8. if (isFirstLayer === void 0) { isFirstLayer = false; }
  9. var conv0 = isFirstLayer
  10. ? extractConvParams(prefix + "/conv0")
  11. : extractSeparableConvParams(prefix + "/conv0");
  12. var conv1 = extractSeparableConvParams(prefix + "/conv1");
  13. var conv2 = extractSeparableConvParams(prefix + "/conv2");
  14. return { conv0: conv0, conv1: conv1, conv2: conv2 };
  15. }
  16. function extractDenseBlock4Params(prefix, isFirstLayer) {
  17. if (isFirstLayer === void 0) { isFirstLayer = false; }
  18. var conv0 = isFirstLayer
  19. ? extractConvParams(prefix + "/conv0")
  20. : extractSeparableConvParams(prefix + "/conv0");
  21. var conv1 = extractSeparableConvParams(prefix + "/conv1");
  22. var conv2 = extractSeparableConvParams(prefix + "/conv2");
  23. var conv3 = extractSeparableConvParams(prefix + "/conv3");
  24. return { conv0: conv0, conv1: conv1, conv2: conv2, conv3: conv3 };
  25. }
  26. return {
  27. extractDenseBlock3Params: extractDenseBlock3Params,
  28. extractDenseBlock4Params: extractDenseBlock4Params
  29. };
  30. }
  31. //# sourceMappingURL=loadParamsFactory.js.map