更新 util/ModelUtils.py

This commit is contained in:
zhoushuliang 2025-07-04 13:49:34 +08:00
parent 3c54c22b68
commit 7a2616df6a
1 changed files with 2 additions and 2 deletions

View File

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