Functions | Variables
deep_sort_app Namespace Reference

Functions

def create_detections (detection_mat, frame_idx, min_height=0)
 
def gather_sequence_info (sequence_dir, detection_file)
 
def parse_args ()
 
def run (sequence_dir, detection_file, output_file, min_confidence, nms_max_overlap, min_detection_height, max_cosine_distance, nn_budget, display)
 

Variables

 args = parse_args()
 

Function Documentation

def deep_sort_app.create_detections (   detection_mat,
  frame_idx,
  min_height = 0 
)
Create detections for given frame index from the raw detection matrix.

Parameters
----------
detection_mat : ndarray
    Matrix of detections. The first 10 columns of the detection matrix are
    in the standard MOTChallenge detection format. In the remaining columns
    store the feature vector associated with each detection.
frame_idx : int
    The frame index.
min_height : Optional[int]
    A minimum detection bounding box height. Detections that are smaller
    than this value are disregarded.

Returns
-------
List[tracker.Detection]
    Returns detection responses at given frame index.

Definition at line 96 of file deep_sort_app.py.

def deep_sort_app.gather_sequence_info (   sequence_dir,
  detection_file 
)
Gather sequence information, such as image filenames, detections,
groundtruth (if available).

Parameters
----------
sequence_dir : str
    Path to the MOTChallenge sequence directory.
detection_file : str
    Path to the detection file.

Returns
-------
Dict
    A dictionary of the following sequence information:

    * sequence_name: Name of the sequence
    * image_filenames: A dictionary that maps frame indices to image
      filenames.
    * detections: A numpy array of detections in MOTChallenge format.
    * groundtruth: A numpy array of ground truth in MOTChallenge format.
    * image_size: Image size (height, width).
    * min_frame_idx: Index of the first frame.
    * max_frame_idx: Index of the last frame.

Definition at line 17 of file deep_sort_app.py.

def deep_sort_app.parse_args ( )
Parse command line arguments.

Definition at line 215 of file deep_sort_app.py.

def deep_sort_app.run (   sequence_dir,
  detection_file,
  output_file,
  min_confidence,
  nms_max_overlap,
  min_detection_height,
  max_cosine_distance,
  nn_budget,
  display 
)
Run multi-target tracker on a particular sequence.

Parameters
----------
sequence_dir : str
    Path to the MOTChallenge sequence directory.
detection_file : str
    Path to the detections file.
output_file : str
    Path to the tracking output file. This file will contain the tracking
    results on completion.
min_confidence : float
    Detection confidence threshold. Disregard all detections that have
    a confidence lower than this value.
nms_max_overlap: float
    Maximum detection overlap (non-maxima suppression threshold).
min_detection_height : int
    Detection height threshold. Disregard all detections that have
    a height lower than this value.
max_cosine_distance : float
    Gating threshold for cosine distance metric (object appearance).
nn_budget : Optional[int]
    Maximum size of the appearance descriptor gallery. If None, no budget
    is enforced.
display : bool
    If True, show visualization of intermediate tracking results.

Definition at line 131 of file deep_sort_app.py.

Variable Documentation

deep_sort_app.args = parse_args()

Definition at line 253 of file deep_sort_app.py.



jsk_perception
Author(s): Manabu Saito, Ryohei Ueda
autogenerated on Mon May 3 2021 03:03:27