otx.algo.detection.heads

Custom head implementations for detection task.

Classes

ATSSHead(model_name, num_classes, ...[, ...])	ATSSHead factory for detection.
RTDETRTransformer(model_name, num_classes[, ...])	RTDETRTransformer factory for detection.
RTMDetSepBNHead(model_name, num_classes, ...)	RTMDetSepBNHead factory for detection.
SSDHead(model_name, num_classes, ...[, test_cfg])	SSDHead factory for detection.
YOLOHead(model_name, num_classes)	YOLOHead factory for detection.
YOLOXHead(model_name, num_classes, train_cfg)	YOLOXHead factory for detection.

class otx.algo.detection.heads.ATSSHead(model_name: str, num_classes: int, anchor_generator: BasePriorGenerator, bbox_coder: BaseBBoxCoder, train_cfg: dict, test_cfg: dict | None = None)

Bases: object

ATSSHead factory for detection.

Constructor for ATSSHead.

class otx.algo.detection.heads.RTDETRTransformer(model_name: str, num_classes: int, eval_spatial_size: tuple[int, int] | None = None)

Bases: object

RTDETRTransformer factory for detection.

Constructor for RTDETRTransformer.

class otx.algo.detection.heads.RTMDetSepBNHead(model_name: str, num_classes: int, anchor_generator: BasePriorGenerator, bbox_coder: BaseBBoxCoder, train_cfg: dict, test_cfg: dict | None = None)

Bases: object

RTMDetSepBNHead factory for detection.

Constructor for RTMDetSepBNHead.

class otx.algo.detection.heads.SSDHead(model_name: str, num_classes: int, anchor_generator: BasePriorGenerator, bbox_coder: BaseBBoxCoder, init_cfg: dict, train_cfg: dict, test_cfg: dict | None = None)

Bases: object

SSDHead factory for detection.

Constructor for SSDHead.

class otx.algo.detection.heads.YOLOHead(model_name: str, num_classes: int)

Bases: object

YOLOHead factory for detection.

Constructor for YOLOHead for v7 and v9.

class otx.algo.detection.heads.YOLOXHead(model_name: str, num_classes: int, train_cfg: dict, test_cfg: dict | None = None)

Bases: object

YOLOXHead factory for detection.

Constructor for YOLOXHead.