otx.algorithms.detection.adapters.mmdet.models.dense_heads#

Initial file for mmdetection dense heads.

Classes

`MMOVRPNHead`(model_path_or_model[, ...])	MMOVRPNHead class for OTX.
`MMOVSSDHead`(model_path_or_model[, ...])	MMOVSSDHead class for OTX.
`MMOVYOLOV3Head`(model_path_or_model[, ...])	MMOVYOLOV3Head class for OTX.

class otx.algorithms.detection.adapters.mmdet.models.dense_heads.MMOVRPNHead(model_path_or_model: str | Model, weight_path: str | None = None, inputs: Dict[str, str | List[str]] | List[str] | str | None = None, outputs: Dict[str, str | List[str]] | List[str] | str | None = None, init_weight: bool = False, verify_shape: bool = True, transpose_cls: bool = False, transpose_reg: bool = False, *args, **kwargs)[source]#

Bases: RPNHead

MMOVRPNHead class for OTX.

Initialize BaseModule, inherited from torch.nn.Module

forward_single(x)[source]#: Forward funtion for MMOVRPNHead.

init_weights()[source]#: Initial weight function of MMOVRPNHead.

class otx.algorithms.detection.adapters.mmdet.models.dense_heads.MMOVSSDHead(model_path_or_model: str | Model, weight_path: str | None = None, inputs: Dict[str, str | List[str]] | List[str] | str | None = None, outputs: Dict[str, str | List[str]] | List[str] | str | None = None, init_weight: bool = False, verify_shape: bool = True, transpose_cls: bool = False, transpose_reg: bool = False, background_index: int | None = None, *args, **kwargs)[source]#

Bases: SSDHead

MMOVSSDHead class for OTX.

Initialize BaseModule, inherited from torch.nn.Module

forward(feats)[source]#: Forward function for MMOVSSDHead.

init_weights()[source]#: Initialize weights function of MMOVSSDHead.

class otx.algorithms.detection.adapters.mmdet.models.dense_heads.MMOVYOLOV3Head(model_path_or_model: str | Model, weight_path: str | None = None, inputs: Dict[str, str | List[str]] | List[str] | str | None = None, outputs: Dict[str, str | List[str]] | List[str] | str | None = None, init_weight: bool = False, verify_shape: bool = True, *args, **kwargs)[source]#

Bases: YOLOV3Head

MMOVYOLOV3Head class for OTX.

Initialize BaseModule, inherited from torch.nn.Module

init_weights()[source]#: Initialize weights of MMOVYOLOV3Head.