(conf, model_output=None, org_model_dir=None)
| 384 | |
| 385 | |
| 386 | def normalize_model_config(conf, model_output=None, org_model_dir=None): |
| 387 | conf = normalize_graph_config(conf, model_output, org_model_dir) |
| 388 | if ModelKeys.subgraphs in conf: |
| 389 | nor_subgraphs = {} |
| 390 | if isinstance(conf[ModelKeys.subgraphs], list): |
| 391 | nor_subgraph = normalize_graph_config(conf[ModelKeys.subgraphs][0], |
| 392 | model_output, org_model_dir) |
| 393 | conf[ModelKeys.input_tensors] = \ |
| 394 | nor_subgraph[ModelKeys.input_tensors] |
| 395 | conf[ModelKeys.output_tensors] = \ |
| 396 | nor_subgraph[ModelKeys.output_tensors] |
| 397 | if ModelKeys.validation_inputs_data in nor_subgraph: |
| 398 | conf[ModelKeys.validation_inputs_data] = \ |
| 399 | nor_subgraph[ModelKeys.validation_inputs_data] |
| 400 | if ModelKeys.validation_outputs_data in nor_subgraph: |
| 401 | conf[ModelKeys.validation_outputs_data] = \ |
| 402 | nor_subgraph[ModelKeys.validation_outputs_data] |
| 403 | set_default_config_value(nor_subgraph, conf) |
| 404 | nor_subgraphs[ModelKeys.default_graph] = nor_subgraph |
| 405 | else: |
| 406 | for graph_name, subgraph in conf[ModelKeys.subgraphs].items(): |
| 407 | nor_subgraph = normalize_graph_config(subgraph, model_output, |
| 408 | org_model_dir) |
| 409 | set_default_config_value(nor_subgraph, conf) |
| 410 | nor_subgraphs[graph_name] = nor_subgraph |
| 411 | |
| 412 | conf[ModelKeys.subgraphs] = nor_subgraphs |
| 413 | |
| 414 | model_base_conf = copy.deepcopy(conf) |
| 415 | del model_base_conf[ModelKeys.subgraphs] |
| 416 | subgraphs = conf[ModelKeys.subgraphs] |
| 417 | for net_name, subgraph in subgraphs.items(): |
| 418 | net_conf = copy.deepcopy(model_base_conf) |
| 419 | net_conf.update(subgraph) |
| 420 | subgraphs[net_name] = net_conf |
| 421 | |
| 422 | MaceLogger.summary(conf) |
| 423 | return conf |
| 424 | |
| 425 | |
| 426 | def find_input_tensors_info(subgraphs, tensor_names): |
no test coverage detected