INNER CODE UNIT · Python
IDs
GeekAlexis/FastMOT · fastmot/detector.py:61
Class IDs to detect. Note class ID starts at zero.
model : str, optional
SSD model to use.
Must be the name of a class that inherits `models.SSD`.
tile_overlap : float, optional
Ratio of overlap to width and height of each tile.
tiling_grid : tuple, optional
Width and height of tile layout to split each frame for batch inference.
conf_thresh : float, optional
Detection confidence threshold.
merge_thresh : float, optional
Overlap threshold to merge bounding boxes across tiles.
max_area : int, optional
Max area of bounding boxes to detect.
"""
super().__init__(size)
self.model = models.SSD.get_model(model)
assert 0 <= tile_overlap <= 1