ssd_hparams

YAHP interface for SSD.

Hparams

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

SSDHparams

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')

Bases: composer.models.model_hparams.ModelHparams

YAHP interface for SSD.

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