YAHP interface for SSD.


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


YAHP interface for SSD.

class composer.models.ssd.ssd_hparams.SSDHparams(initializers=<factory>, num_classes=80, input_size=300, overlap_threshold=0.5, nms_max_detections=200, data='/localdisk/coco')[source]#

Bases: composer.models.model_hparams.ModelHparams

YAHP interface for SSD.

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