22 lines
583 B
Python
22 lines
583 B
Python
"""统一推理引擎抽象层
|
||
|
||
每个引擎实现 BaseEngine 接口，由 EngineFactory 按 inference_engine 名分发。
|
||
所有引擎对外的 detect() 输入为 BGR frame，输出为 list[DetectionResult]，
|
||
与 CameraPipeline 兼容。
|
||
|
||
支持引擎：
|
||
- yolo_pytorch (Yolo-PyTorch，主引擎)
|
||
- onnxruntime
|
||
- openvino
|
||
"""
|
||
|
||
from app.analysis.engines.base import BaseEngine, EngineNotAvailableError, DetectionResult
|
||
from app.analysis.engines.factory import EngineFactory
|
||
|
||
__all__ = [
|
||
"BaseEngine",
|
||
"EngineNotAvailableError",
|
||
"DetectionResult",
|
||
"EngineFactory",
|
||
]
|