From 7a2616df6a6b45ceda41b336dfb897d2e8e3c21a Mon Sep 17 00:00:00 2001 From: zhoushuliang Date: Fri, 4 Jul 2025 13:49:34 +0800 Subject: [PATCH] =?UTF-8?q?=E6=9B=B4=E6=96=B0=20util/ModelUtils.py?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- util/ModelUtils.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/util/ModelUtils.py b/util/ModelUtils.py index 49478f1..9154274 100644 --- a/util/ModelUtils.py +++ b/util/ModelUtils.py @@ -608,8 +608,8 @@ MODEL_CONFIG = { ), # 加载交通模型 ModelType.TRAFFICFORDSJ_FARM_MODEL.value[1]: ( - lambda x, y, r, t, z, h: OneModel(x, y, r, ModelType.TRAFFIC_FARM_MODEL, t, z, h), - ModelType.TRAFFIC_FARM_MODEL, + lambda x, y, r, t, z, h: OneModel(x, y, r, ModelType.TRAFFICFORDSJ_FARM_MODEL, t, z, h), + ModelType.TRAFFICFORDSJ_FARM_MODEL, lambda x, y, z: one_label(x, y, z), lambda x: model_process(x) ),