Skip to content

AnchorFreeDecoupledHead

Anchor-free decoupled detection head based on YOLOX: Exceeding YOLO Series in 2021.

We provide the head of YOLOX as AnchorFreeDecoupledHead. There are no differnece with the original model, and currently, it is set to pass non-maximum suppression function.

Compatibility matrix

Supporting backbones Supporting necks torch.fx NetsPresso
ResNet
MobileNetV3
MixNet
CSPDarkNet
MobileViT
MixTransformer
EfficientFormer 		FPN
YOLOPAFPN
 Supported Supported

Field list

Field Description
name (str) Name must be "yolox_head" to use YOLOX head.
params.act_type (float) Activation function for the head.
params.score_thresh (float) Score thresholding value applied during the decoding step.
params.class_agnostic (bool) Whether to process class-agnostic non-maximum suppression.

Model configuration example

Anchor-free decoupled detection head 
model:
  architecture:
    head:
      name: anchor_free_decoupled_head
      params:
        act_type: "silu"
        # postprocessor - decode
        score_thresh: 0.7
        # postprocessor - nms
        nms_thresh: 0.45
        class_agnostic: False