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

Neck list of mmdetection adapters.

Classes

MMOVFPN(model_path_or_model[, weight_path, ...])

MMOVFPN class for OMZ models.

MMOVSSDNeck(model_path_or_model[, ...])

MMOVSSDNeck class for OMZ models.

MMOVYOLOV3Neck(model_path_or_model[, ...])

MMOVYOLOV3Neck class for OMZ models.

class otx.algorithms.detection.adapters.mmdet.models.necks.MMOVFPN(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: FPN

MMOVFPN class for OMZ models.

Initialize BaseModule, inherited from torch.nn.Module

init_weights(pretrained=None)[source]#

Initial weights function of MMOVFPN.

class otx.algorithms.detection.adapters.mmdet.models.necks.MMOVSSDNeck(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)[source]#

Bases: SSDNeck

MMOVSSDNeck class for OMZ models.

Initialize BaseModule, inherited from torch.nn.Module

init_weights(pretrained=None)[source]#

Initial weights of MMOVSSDNeck.

class otx.algorithms.detection.adapters.mmdet.models.necks.MMOVYOLOV3Neck(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)[source]#

Bases: YOLOV3Neck, ParserMixin

MMOVYOLOV3Neck class for OMZ models.

Initialize BaseModule, inherited from torch.nn.Module

init_weights(pretrained=None)[source]#

Initial weights of MMOVYOLOV3Neck.