models.yolo.Yolo

class Yolo

Bases: BaseConfig

Generates a model similar to yolo8

See https://viso.ai/deep-learning/yolov8-guide/.

Methods

backbone_cfgs	Generates and returns a network backbone configuration
head_cfgs	Generates and returns a network head configuration
neck_cfgs	Generates and returns a network neck configuration

backbone_cfgs() → ListGen

Generates and returns a network backbone configuration

Returns:

Backbone configuration.

Return type:

ListGen

head_cfgs(box_out: int, cls_out: int) → ListGen

Generates and returns a network head configuration

Parameters:

• box_out (int) – Number of output channels for box predictions.

• cls_out (int) – Number of output channels for class predictions.

Returns:

Head configuration.

Return type:

ListGen

neck_cfgs() → ListGen

Generates and returns a network neck configuration

Returns:

Neck configuration.

Return type:

ListGen