From bea36e2601876363d769df60b7f19fca79b37c81 Mon Sep 17 00:00:00 2001 From: th Date: Sat, 23 Aug 2025 17:08:47 +0800 Subject: [PATCH] =?UTF-8?q?1=EF=BC=89=E8=87=AA=E7=A0=94=E8=BD=A6=E7=89=8C?= =?UTF-8?q?=E6=A8=A1=E5=9E=8B=E6=94=AF=E6=8C=81engine?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- enums/ModelTypeEnum.py | 6 ++++-- util/ModelUtils.py | 2 -- 2 files changed, 4 insertions(+), 4 deletions(-) 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)