p47068915
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

fix onnx resize parser

This commit is contained in:
xuanyue 2020-10-27 22:08:33 +08:00
parent acd156c084
commit b1b4165bdc
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
mindspore/lite/tools/converter/parser/onnx/onnx_resize_parser.cc
View File

@ -55,7 +55,7 @@ STATUS OnnxResizeParser::Parse(const onnx::GraphProto &onnx_graph, const onnx::N
{"tf_half_pixel_for_nn", schema::CoordinateTransformMode_TF_HALF_PIXEL},
{"tf_crop_and_resize", schema::CoordinateTransformMode_TF_CROP_AND_RESIZE},
};
return transform_map[onnx_node_attr.strings(0)];
return transform_map[onnx_node_attr.s()];
}();
} else if (attribute_name == "cubic_coeff_a") {
attr->cubicCoeff = onnx_node_attr.f();
@ -70,7 +70,7 @@ STATUS OnnxResizeParser::Parse(const onnx::GraphProto &onnx_graph, const onnx::N
{"linear", schema::ResizeMethod_LINEAR},
{"cubic", schema::ResizeMethod_CUBIC},
};
return resize_mode[onnx_node_attr.strings(0)];
return resize_mode[onnx_node_attr.s()];
}();
} else if (attribute_name == "nearest_mode") {
attr->nearestMode = [&]() {
@ -80,7 +80,7 @@ STATUS OnnxResizeParser::Parse(const onnx::GraphProto &onnx_graph, const onnx::N
{"floor", schema::NearestMode_FLOOR},
{"ceil", schema::NearestMode_CEIL},
};
return nearest_mode[onnx_node_attr.strings(0)];
return nearest_mode[onnx_node_attr.s()];
}();
}
}