diff --git a/enums/ModelTypeEnum.py b/enums/ModelTypeEnum.py index 392145c..ff36775 100644 --- a/enums/ModelTypeEnum.py +++ b/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) }, }, diff --git a/util/ModelUtils.py b/util/ModelUtils.py index d3e9f7b..2ee4dc8 100644 --- a/util/ModelUtils.py +++ b/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)