HRNetStageConfig

class plugins.extract.align.hrnet.HRNetStageConfig(num_modules: int, num_branches: int, num_blocks: list[int], num_channels: list[int], block: Literal['ATTENTION', 'BLOCK', 'BOTTLENECK'])

Bases: object

Configuration settings for each stage of HRNet

Parameters:

• num_modules (int)

• num_branches (int)

• num_blocks (list[int])

• num_channels (list[int])

• block (Literal['ATTENTION', 'BLOCK', 'BOTTLENECK'])