models.generator.BaseConfig

class BaseConfig

Bases: object

Base class for model configuration generators

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