未验证 提交 33f29a22 编写于 作者: 萝卜菜 提交者: GitHub

fix pytengine example error (#1279)

* apply code-format changes

* fix python example error
上级 159411b6
......@@ -50,6 +50,7 @@ def main(args):
data = ((data - img_mean) * scale).astype(np.float32)
data = np.ascontiguousarray(data.transpose((2, 0, 1)))
assert data.dtype == np.float32
data = data.copy()
graph = tg.Graph(None, 'tengine', tm_file)
input_tensor = graph.getInputTensor(0, 0)
......
......@@ -60,6 +60,7 @@ def main(args):
data = ((data - img_mean) * scale[0]).astype(np.float32)
data = data.transpose((2, 0, 1))
assert data.dtype == np.float32
data = data.copy()
graph = tg.Graph(None, 'tengine', tm_file)
input_tensor = graph.getInputTensor(0, 0)
......
......@@ -63,6 +63,7 @@ def main(args):
data = ((data - img_mean) * DEFAULT_SCALE).astype(np.float32)
data = data.transpose((2, 0, 1))
assert data.dtype == np.float32
data = data.copy()
graph = tg.Graph(None, 'tengine', tm_file)
input_tensor = graph.getInputTensor(0, 0)
......
......@@ -322,7 +322,7 @@ def main(args):
#image_data = image_data.transpose((2, 0, 1)).astype(np.float32)
image_data = np.ascontiguousarray(image_data.transpose((2, 0, 1)).astype(np.float32))
image_data = image_data.copy()
#print("img_h, img_w, img_c: %d, %d, %d" %(img_h, img_w, img_c))
input_tensor = graph.getTensorByName(input_name)
......
......@@ -2132,9 +2132,7 @@ static int load_resize(ir_graph_t* graph, ir_node_t* node, const onnx::NodeProto
interp_param->width_scale = 0;
std::string coordinate_transformation_mode = GetAttributeOrDefault<std::string>(onnx_node, "coordinate_transformation_mode", "half_pixel");
TASSERT(coordinate_transformation_mode == "half_pixel" ||
coordinate_transformation_mode == "align_corners" ||
coordinate_transformation_mode == "asymmetric");
TASSERT(coordinate_transformation_mode == "half_pixel" || coordinate_transformation_mode == "align_corners" || coordinate_transformation_mode == "asymmetric");
int align_corner = (coordinate_transformation_mode == "align_corners");
if (onnx_node.input_size() == 1)
......
......@@ -749,7 +749,6 @@ tm_uoffset_t SaveTmMinimumOp(void* const start_ptr, tm_uoffset_t* cur_pos, ir_no
return WriteTmObject(start_ptr, cur_pos, &tm_op, sizeof(TM2_Operator));
}
tm_uoffset_t SaveTmSqueezeOp(void* const start_ptr, tm_uoffset_t* cur_pos, ir_node_t* node)
{
struct squeeze_param* p = (struct squeeze_param*)node->op.param_mem;
......@@ -1608,7 +1607,7 @@ op_save_t SaveTmOpFunc(uint32_t op_type)
case OP_MINIMUM:
return SaveTmMinimumOp;
default:
fprintf(stderr, "Operator #%d not supported in tengine model yet\n",op_type);
fprintf(stderr, "Operator #%d not supported in tengine model yet\n", op_type);
return nullptr;
}
}
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册