31 lines
1.0 KiB
Python
31 lines
1.0 KiB
Python
|
"""向后兼容 shim：保留 ObjectDetector 名称，内部委托给 OnnxEngine
|
|||
|
|
|
|||
|
|
旧代码 `from app.analysis.detector import ObjectDetector` 仍可工作，
|
|||
|
|
新代码请直接使用 `app.analysis.engines.EngineFactory`。
|
|||
|
|
"""
|
|||
|
|
import logging
|
|||
|
|
|
|||
|
|
from app.analysis.engines.onnx_engine import OnnxEngine
|
|||
|
|
|
|||
|
|
logger = logging.getLogger("analysis.detector")
|
|||
|
|
|
|||
|
|
|
|||
|
|
class ObjectDetector(OnnxEngine):
|
|||
|
|
"""已废弃：等价于 OnnxEngine（detect 任务）。保留以兼容旧 import。"""
|
|||
|
|
|
|||
|
|
def __init__(self, model_path=None, labels=None, input_size=(640, 640),
|
|||
|
|
conf_threshold=0.4, iou_threshold=0.5, providers=None):
|
|||
|
|
super(ObjectDetector, self).__init__(
|
|||
|
|
model_file=model_path or "",
|
|||
|
|
labels=labels,
|
|||
|
|
input_size=input_size,
|
|||
|
|
conf_threshold=conf_threshold,
|
|||
|
|
iou_threshold=iou_threshold,
|
|||
|
|
providers=providers,
|
|||
|
|
task_type="detect",
|
|||
|
|
device="cpu",
|
|||
|
|
algorithm_type="yolo8",
|
|||
|
|
)
|
|||
|
|
if model_path:
|
|||
|
|
self.load()
|