sportslabkit.detection_model.yolov8

Overview

Classes

YOLOv8	YOLO model wrapper.
YOLOv8n	YOLO model wrapper.
YOLOv8s	YOLO model wrapper.
YOLOv8m	YOLO model wrapper.
YOLOv8l	YOLO model wrapper.
YOLOv8x	YOLO model wrapper.

Classes

class sportslabkit.detection_model.yolov8.YOLOv8(model: str = '', agnostic_nms: bool = False, multi_label: bool = False, classes: list[str] | None = None, max_det: int = 1000, amp: bool = False, imgsz: int = 640, conf: float = 0.25, iou: float = 0.45, device: str = 'cpu', verbose: bool = False, augment: bool = False)

Bases: sportslabkit.detection_model.base.BaseDetectionModel

YOLO model wrapper. Receives the arguments controlling inference as ‘inference_config’ when initialized.

Overview

Attributes

hparam_search_space

-

Methods

forward(x, **kwargs)

param x: input tensor

Members

hparam_search_space

forward(x, **kwargs)

Parameters:

x (Tensor) – input tensor

Returns:

output tensor

Return type:

Tensor

class sportslabkit.detection_model.yolov8.YOLOv8n(**model_config)

Bases: YOLOv8

YOLO model wrapper. Receives the arguments controlling inference as ‘inference_config’ when initialized.

class sportslabkit.detection_model.yolov8.YOLOv8s(**model_config)

Bases: YOLOv8

YOLO model wrapper. Receives the arguments controlling inference as ‘inference_config’ when initialized.

class sportslabkit.detection_model.yolov8.YOLOv8m(**model_config)

Bases: YOLOv8

YOLO model wrapper. Receives the arguments controlling inference as ‘inference_config’ when initialized.

class sportslabkit.detection_model.yolov8.YOLOv8l(**model_config)

Bases: YOLOv8

YOLO model wrapper. Receives the arguments controlling inference as ‘inference_config’ when initialized.

class sportslabkit.detection_model.yolov8.YOLOv8x(model: str = 'yolov8x', agnostic_nms: bool = False, multi_label: bool = False, classes: list[str] | None = None, max_det: int = 1000, amp: bool = False, imgsz: int = 640, conf: float = 0.25, iou: float = 0.45, device: str = 'cpu', verbose: bool = False, augment: bool = False)

Bases: YOLOv8

YOLO model wrapper. Receives the arguments controlling inference as ‘inference_config’ when initialized.