Skip to content

PIDNet

PIDNet model based on PIDNet: A Real-time Semantic Segmentation Network Inspired by PID Controllers.

Compatibility matrix

Supporting Task(s) torch.fx NetsPresso
Segmentation Supported Supported

Field list

Field Description
name (str) Name must be "pidnet" to use PIDNet model.
m (int) Residual block repetetion count for 1, 2 stage of I branch, and 3, 4 stage of P branch.
n (int) Residual block repetition count for 3, 4 stage of I branch
channels (int) Base dimension of the overall model except ppm and head module.
ppm_channels (int) Dimension of the ppm module.
head_channels (int) Dimension of the head module.
PIDNet-s 
model:
  architecture:
    full:
      name: pidnet
      m: 2
      n: 3
      channels: 32
      ppm_channels: 96
      head_channels: 128
PIDNet-m 
model:
  architecture:
    full:
      name: pidnet
      m: 2
      n: 3
      channels: 64
      ppm_channels: 96
      head_channels: 128
PIDNet-l 
model:
  architecture:
    full:
      name: pidnet
      m: 3
      n: 4
      channels: 64
      ppm_channels: 112
      head_channels: 256