提交新模型

LPRNet_ORT_infer.py

@@ -26,7 +26,7 @@ def LPRNetPreprocess(image):
     return img
 
 def LPRNetPostprocess(infer_res):
-    preb_label = np.argmax(infer_res, axis=0)
+    preb_label = np.argmax(infer_res, axis=1)[0]
     no_repeat_blank_label = []
     pre_c = preb_label[0]
     if pre_c != len(CHARS) - 1:

LPRNet_migraphx_infer.py

@@ -28,7 +28,7 @@ def LPRNetPreprocess(image):
     return img
 
 def LPRNetPostprocess(infer_res):
-    preb_label = np.argmax(infer_res, axis=0)
+    preb_label = np.argmax(infer_res, axis=1)[0]
     no_repeat_blank_label = []
     pre_c = preb_label[0]
     if pre_c != len(CHARS) - 1:

train.py

@@ -240,7 +240,7 @@ def get_parser():
     parser.add_argument('--weight_decay', default=2e-5, type=float, help='Weight decay for SGD')
     parser.add_argument('--lr_schedule', default=[4, 8, 12, 14, 16], help='schedule for learning rate.')
     parser.add_argument('--save_folder', default='./weights/', help='Location to save checkpoint models')
-    parser.add_argument('--pretrained_model', default='./weights/Final_LPRNet_model.pth', help='pretrained base model')
+    parser.add_argument('--pretrained_model', default='model/lprnet.pth', help='pretrained base model')
     args = parser.parse_args()
     return args