更新室内导盲代码

This commit is contained in:
Kevin Wong
2026-01-05 14:39:16 +08:00
parent baf9e235a1
commit 122c07b28e
5 changed files with 22 additions and 504 deletions

View File

@@ -483,7 +483,7 @@ def load_indoor_model():
from model_utils import is_tensorrt_engine # Imported here for usage
try:
indoor_model_path = os.getenv("INDOOR_MODEL", "model/yolo11l-seg-indoor14.pt")
indoor_model_path = os.getenv("INDOOR_MODEL", "model/yolo11l-seg-indoor.engine")
# 优先使用 TensorRT 引擎
indoor_model_path = get_best_model_path(indoor_model_path)
print(f"[INDOOR] 尝试加载室内导盲模型: {indoor_model_path}")