修改代码

LPRNet_migraphx_infer.py

@@ -49,11 +49,10 @@ def LPRNetInference(model_name, imgs):
     if model_name[-3:] == 'mxr':
         model = migraphx.load(model_name)
     else:
+        print('convert onnx to mxr...')
         model = migraphx.parse_onnx(model_name)
-        # migraphx.quantize_fp16(model)
         model.compile(t=migraphx.get_target("gpu"),device_id=0) # device_id: 设置GPU设备，默认为0号设备(>=1.2版本中支持)
-        path = model_name.split('/')
-        migraphx.save(model, '/'.join((path[:-1], path[-1][-4]+'mxr')))
+        migraphx.save(model, 'model/LPRNet.mxr')
 
     inputName=model.get_parameter_names()[0]
     inputShape=model.get_parameter_shapes()[inputName].lens()
@@ -65,8 +64,8 @@ def LPRNetInference(model_name, imgs):
     return result
 
 if __name__ == '__main__':
-    model_name = 'LPRNet.onnx'
-    # model_name = 'model/LPRNet.mxr'
+    # model_name = 'model/LPRNet.onnx'
+    model_name = 'model/LPRNet.mxr'
     image = 'imgs/川JK0707.jpg'
     InferRes = LPRNetInference(model_name, image)
     print(image, 'Inference Result:', InferRes)