modify

README.md

@@ -4,6 +4,8 @@
 
 [yolov7 车牌检测+识别](https://github.com/we0091234/yolov7_plate)
 
+[安卓NCNN](https://github.com/Ayers-github/Chinese-License-Plate-Recognition)
+
 ## **最全车牌识别算法，支持12种中文车牌类型**
 
 **环境要求: python >=3.6  pytorch >=1.7**

detect_plate.py

@@ -231,10 +231,10 @@ def get_second(capture):
 
 if __name__ == '__main__':
     parser = argparse.ArgumentParser()
-    parser.add_argument('--detect_model', nargs='+', type=str, default='runs/train/exp40/weights/best.pt', help='model.pt path(s)')  #检测模型
+    parser.add_argument('--detect_model', nargs='+', type=str, default='weights/plate_detect.pt', help='model.pt path(s)')  #检测模型
     parser.add_argument('--rec_model', type=str, default='weights/plate_rec_color.pth', help='model.pt path(s)')#车牌识别+颜色识别模型
     parser.add_argument('--is_color',type=bool,default=True,help='plate color')     #是否识别颜色
-    parser.add_argument('--image_path', type=str, default='gangao/Test', help='source') 
+    parser.add_argument('--image_path', type=str, default='imgs', help='source') 
     parser.add_argument('--img_size', type=int, default=640, help='inference size (pixels)')
     parser.add_argument('--output', type=str, default='result1', help='source') 
     parser.add_argument('--video', type=str, default='', help='source') 

models/yolo.py

@@ -45,6 +45,7 @@ def __init__(self, nc=80, anchors=(), ch=()):  # detection layer
     def forward(self, x):
         # x = x.copy()  # for profiling
         z = []  # inference output
+        # self.training=True
         if self.export_cat:
             for i in range(self.nl):
                 x[i] = self.m[i](x[i])  # conv