modules.vision.detection.models.yolo_model

Classes Diagram

Module Dependencies

Classes

YOLOModel

General YOLO model wrapper for v8, v11, etc.

Constructor

YOLOModel(self, model_path: Path)

Members

model

Methods

track(self, frame)

Run tracking/detection on a frame.

predict(self, frame)

Attributes

logger