修改readme

README.md

@@ -55,6 +55,7 @@ LPR模型用test.py对训练出的模型进行测试，使用方法如下：
     python test.py \
         --model 需要测试的pth模型路径 \
         --imgpath 测试集路径(文件夹或图像皆可)  \
+        --batch_size 测试时的batch size大小 \
         --export_onnx True/False(该参数用于选择是否需要将pth模型转为onnx模型) \
         --dynamic True/False(该参数用于选择onnx模型是否使用动态的batch size)
 

lprnet.py

@@ -89,5 +89,3 @@ def build_lprnet(class_num, phase=False):
         return Net.train()
     else:
         return Net.eval()
-
-

train.py

@@ -178,6 +178,7 @@ def train(args):
 
         if (iteration + 1) % args.test_interval == 0:
             Greedy_Decode_Eval(lprnet, test_dataset, args)
+            # lprnet.train()
 
         start_time = time.time()
         # load train data