Functions | |
| def | create_detections |
| def | gather_sequence_info |
| def | parse_args |
| def | run |
Variables | |
| tuple | args = parse_args() |
| 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 129 of file deep_sort_app.py.
| tuple deep_sort_app::args = parse_args() |
Definition at line 253 of file deep_sort_app.py.