1）自研车牌模型支持engine

enums/ModelTypeEnum.py

@@ -889,16 +889,18 @@ class ModelType(Enum):
         'rainbows': COLOR,
         'models': [
             {
-                'weight': '../weights/pth/AIlib2/carplate/plate_yolov5s_v3.jit',
+                #'weight': '../weights/pth/AIlib2/carplate/plate_yolov5s_v3.jit',
+                'weight': '../weights/trt/AIlib2/carplate/yolov5_%s_fp16.engine' % (gpuName),
                 'name': 'yolov5',
                 'model': yolov5Model,
                 'par': {
-                    'trtFlag_det': False,
+                    'trtFlag_det': True,
                     'device': 'cuda:0',
                     'half': True,
                     'conf_thres': 0.4,
                     'iou_thres': 0.45,
                     'nc': 1,
+                    'plate':8,
                     'plate_dilate': (0.5, 0.1)
                 },
             },

util/ModelUtils.py

@@ -390,8 +390,6 @@ class CARPLATEModel:
             detpar = par['models'][0]['par']
             # new_device = torch.device(par['device'])
             # modelList=[ modelPar['model'](weights=modelPar['weight'],par=modelPar['par'])  for  modelPar in par['models'] ]
-            logger.info("########################加载 plate_yolov5s_v3.jit 成功 ########################, requestId:{}",
-                        requestId)
             self.model_conf = (modeType, device, modelList, detpar, par['rainbows'])
         except Exception:
             logger.error("模型加载异常：{}, requestId:{}", format_exc(), requestId)