mobileone_params

Typed parameters of the mobileone family; importable without Keras.

MobileOneBlockParams()

MobileOne block parameters

model_config = ConfigDict(frozen=True, extra='forbid')

filters: int = Field(..., description='# filters')

Number of filters

depth: int = Field(default=1, description='Layer depth')

Layer depth

kernel_size: int | tuple[int, int] = Field(default=3, description='Kernel size')

Kernel size

strides: int | tuple[int, int] = Field(default=1, description='Stride size')

Stride size

padding: int | tuple[int, int] = Field(default=0, description='Padding size')

Padding size

se_ratio: float = Field(default=8, description='Squeeze Excite ratio')

Squeeze Excite ratio

se_depth: int = Field(default=0, description='Depth length to apply SE')

Depth length to apply SE

num_conv_branches: int = Field(default=2, description='# conv branches')

Number of conv branches

activation: str = Field(default='relu6', description='Activation function')

Activation function

MobileOneParams()

MobileOne parameters

model_config = ConfigDict(frozen=True, extra='forbid')

family: Literal['mobileone'] = 'mobileone'

Model family

blocks: list[MobileOneBlockParams] = Field(default_factory=list, description='MobileOne blocks')

MobileOne blocks

input_filters: int = Field(default=3, description='Input filters')

Input filters

input_kernel_size: int | tuple[int, int] = Field(default=3, description='Input kernel size')

Input kernel size

input_strides: int | tuple[int, int] = Field(default=2, description='Input stride')

Input stride

input_padding: int | tuple[int, int] = Field(default=1, description='Input padding')

Input padding

include_top: bool = Field(default=True, description='Include top')

Include top

num_classes: int | None = Field(default=None, gt=0, description='Classes of the output layer')

Classes of the output layer

output_activation: str | None = Field(default=None, description='Output activation')

Output activation

dropout: float = Field(default=0.2, description='Dropout rate')

Dropout rate