composer.models.ssd.ssd#

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

Classes

SSD

Single Shot Object detection Model with pretrained ResNet34 backbone extending ComposerModel.
class composer.models.ssd.ssd.SSD(input_size, overlap_threshold, nms_max_detections, num_classes, data)[source]#

Bases: composer.models.base.ComposerModel

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

Parameters

  • input_size (int, optional) โ€“ input image size. Default: 300.

  • num_classes (int, optional) โ€“ The number of classes to detect. Default: 80.

  • overlap_threshold (float, optional) โ€“ Minimum IOU threshold for NMS. Default: 0.5.

  • nms_max_detections (int, optional) โ€“ Max number of boxes after NMS. Default: 200.

  • data (str, optional) โ€“ path to coco dataset. Default: "/localdisk/coco".