SSD

class composer.models.SSD(input_size, overlap_threshold, nms_max_detections, num_classes, data)

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".