INNER CODE UNIT · Python

SSDDetector

GeekAlexis/FastMOT · fastmot/detector.py:45

class SSDDetector(Detector):
    def __init__(self, size,
                 class_ids,
                 model='SSDInceptionV2',
                 tile_overlap=0.25,
                 tiling_grid=(4, 2),
                 conf_thresh=0.5,
                 merge_thresh=0.6,
                 max_area=120000):
        """An object detector for SSD models.

        Parameters
        ----------
        size : tuple
            Width and height of each frame.
        class_ids : sequence
            Class IDs to detect. Note class ID starts at zero.
        model : str, optional

View source record →