unext_params

Typed parameters of the unext family; importable without Keras.

UNextBlockParams()

UNext 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

ddepth: int | None = Field(default=None, description='Layer decoder depth')

Layer decoder depth

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

Kernel size

pool: int | tuple[int, int] = Field(default=2, description='Pool size')

Pool size

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

Stride size

skip: bool = Field(default=True, description='Add skip connection')

Add skip connection

expand_ratio: float = Field(default=1, description='Expansion ratio')

Expansion ratio

se_ratio: float = Field(default=0, description='Squeeze and excite ratio')

Squeeze and excite ratio

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

Dropout rate

norm: Literal['batch', 'layer'] | None = Field(default='layer', description='Normalization type')

Normalization type

UNextParams()

UNext parameters

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

family: Literal['unext'] = 'unext'

Model family

blocks: list[UNextBlockParams] = Field(default_factory=list, description='UNext blocks')

UNext blocks

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; required with include_top')

Classes of the output layer; required with include_top

use_logits: bool = Field(default=True, description='Use logits')

Use logits

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

Output kernel size

output_kernel_stride: int | tuple[int, int] = Field(default=1, description='Output kernel stride')

Output kernel stride