SSDHparams#
- class composer.models.SSDHparams(initializers=<factory>, num_classes=80, input_size=300, overlap_threshold=0.5, nms_max_detections=200, data='/localdisk/coco')[source]#
-
- 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".