#include <ONNXUtils/ONNXWrapper.h>
#include <iostream>
Go to the source code of this file.
|
int | main (int narg, char *argv[]) |
|
◆ main()
int main |
( |
int |
narg, |
|
|
char * |
argv[] |
|
) |
| |
Definition at line 4 of file extract-onnx-metadata.cxx.
6 if (narg != 3 && narg != 2) {
7 std::cout <<
"usage: " <<
argv[0] <<
" <onnx_file> [key]" << std::endl;
17 for (
const auto &
p :
model->GetModelInputs() )
19 std::string separator;
20 std::cout <<
"Input: " <<
p.first <<
", shape: (";
21 for (
auto x :
p.second) {
22 std::cout << separator <<
x;
25 std::cout <<
")" << std::endl;
27 for (
const auto &
p :
model->GetModelOutputs() )
29 std::string separator;
30 std::cout <<
"Outputs: " <<
p.first <<
", shape: (";
31 for (
auto x :
p.second) {
32 std::cout << separator <<
x;
35 std::cout <<
")" << std::endl;
37 for (
const auto &
p :
model->GetMETAData() )
39 std::cout <<
"META data: "<<
p.first <<
'\t' <<
p.second.at(0) << std::endl;
47 std::cout <<
model->GetMETADataByKey(
argv[2]) << std::endl;