修改模型输入数据格式

Python/PaddleOCR_infer_migraphx.py

@@ -454,9 +454,10 @@ class det_rec_functions(object):
         img_part, shape_part_list = data_part
         img_part = np.expand_dims(img_part, axis=0)
         shape_part_list = np.expand_dims(shape_part_list, axis=0)
+        img_part = np.ascontiguousarray(img_part)
 
         # migraphx推理
-        resultDets = self.modelDet.run({self.modelDet.get_parameter_names()[0]: migraphx.argument(img_part)})
+        resultDets = self.modelDet.run({self.modelDet.get_parameter_names()[0]: img_part})
         # 获取第一个输出节点的数据,migraphx.argument类型
         resultDet = resultDets[0]
         outs_part = np.array(resultDet)
@@ -499,7 +500,7 @@ class det_rec_functions(object):
         img = img[np.newaxis, :]
         
         # migraphx推理
-        results = self.modelRec.run({self.modelRec.get_parameter_names()[0]: migraphx.argument(img)})
+        results = self.modelRec.run({self.modelRec.get_parameter_names()[0]: img})
         # 获取第一个输出节点的数据,migraphx.argument类型
         result = results[0]
         outs = np.array(result)