fix infer error model path

infer_axmodel.py

@@ -533,9 +533,9 @@ def text_recognizer(session,img_list,shape=[3,48,320],character_dict_path=r"./pp
 def init_args():
     parser = argparse.ArgumentParser()
     parser.add_argument("--img_path",type=str,default=str(r"./11.jpg"),help="Path to input image.")
-    parser.add_argument("--det_model_dir",type=str,default=str(r"./ax650/det.axmodel"),help="Path to detection model.")
-    parser.add_argument("--rec_model_dir",type=str,default=str(r"./ax650/rec.axmodel"),help="Path to recognition model.")
-    parser.add_argument("--cls_model_dir",type=str,default=str(r"./ax650/cls.axmodel"),help="Path to classification model.")
+    parser.add_argument("--det_model_dir",type=str,default=str(r"./ax650/det_npu3.axmodel"),help="Path to detection model.")
+    parser.add_argument("--rec_model_dir",type=str,default=str(r"./ax650/rec_npu3.axmodel"),help="Path to recognition model.")
+    parser.add_argument("--cls_model_dir",type=str,default=str(r"./ax650/cls_npu3.axmodel"),help="Path to classification model.")
     parser.add_argument("--character_dict_path",type=str,default=str(r"./ppocrv5_dict.txt"),help="recognition dictionary")
     parser.add_argument("--det_limit_side_len", type=float, default=[960,960],help="detection model input size")
     parser.add_argument("--rec_image_shape", type=str, default=[3, 48, 320],help="recognition model input size")
@@ -544,9 +544,9 @@ def init_args():
     return parser.parse_args()
 
 def main(args):
-    det_session = ort.InferenceSession(args.det_model_dir,providers='AxEngineExecutionProvider')
-    rec_session = ort.InferenceSession(args.rec_model_dir,providers='AxEngineExecutionProvider')
-    cls_session = ort.InferenceSession(args.cls_model_dir,providers='AxEngineExecutionProvider')
+    det_session = ort.InferenceSession(args.det_model_dir)
+    rec_session = ort.InferenceSession(args.rec_model_dir)
+    cls_session = ort.InferenceSession(args.cls_model_dir)
 
     image=cv2.imread(args.img_path)
 
@@ -584,4 +584,4 @@ def main(args):
 
 if __name__=='__main__':
     args=init_args()
-    main(args)
+    main(args)