otx.algo.segmentation.heads

Head modules for OTX segmentation model.

Classes

FCNHead(model_name, num_classes)	FCNHead factory for segmentation.
LightHamHead(model_name, num_classes)	LightHamHead factory for segmentation.

class otx.algo.segmentation.heads.FCNHead(model_name: str, num_classes: int)

Bases: object

FCNHead factory for segmentation.

Constructor for FCNHead.

class otx.algo.segmentation.heads.LightHamHead(model_name: str, num_classes: int)

Bases: object

LightHamHead factory for segmentation.

Constructor for FCNHead.