weijuesen
11 lines
507 B
Python
11 lines
507 B
Python
import os
|
|
|
|
# 运行模式：mock（默认，无模型文件）| onnx（加载 best.onnx 真实推理）
|
|
MODEL_MODE = os.getenv("MODEL_MODE", "mock")
|
|
MODEL_PATH = os.getenv("MODEL_PATH", "models/best.onnx")
|
|
# YOLO 类别（二分类训练基线：healthy/sick；7 类病种扩展后再调整）
|
|
MODEL_LABELS = os.getenv("MODEL_LABELS", "healthy,sick").split(",")
|
|
# mock 模式返回的固定结果
|
|
MOCK_CLASS = os.getenv("MOCK_CLASS", "healthy")
|
|
MOCK_CONFIDENCE = float(os.getenv("MOCK_CONFIDENCE", "0.95"))
|