composer.models.ssd#

Modules

composer.models.ssd.base_model

SSD 300 resnet backbones in PyTorch adapted from MLCommons.

composer.models.ssd.ssd

Single Shot Object Detection model with pretrained ResNet34 backbone extending ComposerModel.

composer.models.ssd.ssd300

SSD 300 architecture in PyTorch adapted from MLCommons.

composer.models.ssd.ssd_hparams

YAHP interface for SSD.

composer.models.ssd.utils

SSD 300 utils adapted from MLCommons.

SSD300 for object detection on MSCOCO.

Classes

SSD

Single Shot Object detection Model with pretrained ResNet34 backbone extending ComposerModel.

Hparams

These classes are used with yahp for YAML-based configuration.

SSDHparams

YAHP interface for SSD.