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) ),