(model_file_name)
| 152 | |
| 153 | |
| 154 | def model_output_names(model_file_name): |
| 155 | with open(model_file_name, 'rb') as pfile: |
| 156 | data_str = pfile.read() |
| 157 | model_proto = onnx.ModelProto() |
| 158 | model_proto.ParseFromString(data_str) |
| 159 | output_names = [out.name for out in model_proto.graph.output] |
| 160 | |
| 161 | return output_names |
| 162 | |
| 163 | |
| 164 | def get_input_shapes(sample_case, param_names): |