ssd#
Single Shot Object Detection model with pretrained ResNet34 backbone extending ComposerModel
.
Classes
Single Shot Object detection Model with pretrained ResNet34 backbone extending |
- 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"
.