MCPcopy Create free account

hub / github.com/Zhongdao/Towards-Realtime-MOT / functions

Functions130 in github.com/Zhongdao/Towards-Realtime-MOT

Methodforward
(self, x, targets=None, targets_len=None)
models.py:238
Functionfuse_motion
(kf, cost_matrix, tracks, detections, only_position=False, lambda_=0.98)
tracker/matching.py:98
Functioniou_distance
Compute cost based on IoU :type atracks: list[STrack] :type btracks: list[STrack] :rtype cost_matrix np.ndarray
tracker/matching.py:60
Functionlinear_assignment
(cost_matrix, thresh)
tracker/matching.py:26
Functionload_classes
Loads class labels at 'path'
utils/utils.py:30
Functionmerge_matches
(m1, m2, shape)
tracker/matching.py:9
Functionmodel_info
Prints out a line-by-line description of a PyTorch model ending with a summary.
utils/utils.py:39
Methodmulti_predict
Run Kalman filter prediction step (Vectorized version). Parameters ---------- mean : ndarray The Nx8 dimensi
utils/kalman_filter.py:154
Functionparse_data_cfg
Parses the data configuration file
utils/parse_config.py:24
Functionplot_detections
(image, tlbrs, scores=None, color=(255, 0, 0), ids=None)
utils/visualization.py:68
Functionplot_one_box
Plots one bounding box on image img.
utils/utils.py:54
Functionplot_results
Plot YOLO training results from the file 'results.txt' Example of what this is trying to plot can be found at: https://user-images.gi
utils/utils.py:569
Functionplot_tracking
(image, tlwhs, obj_ids, scores=None, frame_id=0, fps=0., ids2=None)
utils/visualization.py:28
Functionplot_trajectory
(image, tlwhs, track_ids)
utils/visualization.py:57
Functionpooling_nms
(heatmap, kernel=1)
utils/utils.py:419
Methodpredict
Run Kalman filter prediction step. Parameters ---------- mean : ndarray The 8 dimensional mean vector of the
utils/kalman_filter.py:88
Methodpredict
(self)
tracker/basetrack.py:42
Methodpredict
(self)
tracker/multitracker.py:39
Functionread_mot_results
(filename, is_gt, is_ignore)
utils/io.py:63
Functionresize_image
(image, max_size=800)
utils/visualization.py:21
Functionsave_weights
(self, path, cutoff=-1)
models.py:380
Functionshift_tensor_vertically
(t, delta)
models.py:283
Functionstrip_optimizer_from_checkpoint
(filename='weights/best.pt')
utils/utils.py:562
Methodtlbr
Convert bounding box to format `(min x, min y, max x, max y)`, i.e., `(top left, bottom right)`.
tracker/multitracker.py:120
Methodtlwh
Get current position in bounding box format `(top left x, top left y, width, height)`.
tracker/multitracker.py:107
Methodtlwh_to_tlbr
(tlwh)
tracker/multitracker.py:151
Functiontlwhs_to_tlbrs
(tlwhs)
utils/visualization.py:5
Methodupdate
(self, *args, **kwargs)
tracker/basetrack.py:45
Functionweights_init_normal
(m)
utils/utils.py:70
Functionwrite_results
(filename, results_dict: Dict, data_type: str)
utils/io.py:8
← previous101–130 of 130, ranked by callers