MCPcopy Create free account

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

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

↓ 7 callersFunctionmkdir_if_missing
(dir)
utils/utils.py:14
↓ 5 callersFunctionload_darknet_weights
(self, weights, cutoff=-1)
models.py:308
↓ 5 callersFunctionparse_model_cfg
Parses the yolo-v3 layer configuration file and returns module definitions
utils/parse_config.py:1
↓ 4 callersFunctionletterbox
(img, height=608, width=1088, color=(127.5, 127.5, 127.5))
utils/datasets.py:239
↓ 4 callersMethodtlwh_to_xyah
Convert bounding box to format `(center x, center y, aspect ratio, height)`, where the aspect ratio is `width / height`.
tracker/multitracker.py:130
↓ 4 callersMethodupdate
Update a matched track :type new_track: STrack :type frame_id: int :type update_feature: bool :return:
tracker/multitracker.py:84
↓ 3 callersMethod__init__
(self, cfg_dict, nID=0, test_emb=False)
models.py:219
↓ 3 callersFunctionjoint_stracks
(tlista, tlistb)
tracker/multitracker.py:351
↓ 3 callersFunctionnon_max_suppression
Removes detections with lower object confidence score than 'conf_thres' Non-Maximum Suppression to further filter detections. Returns
utils/utils.py:425
↓ 3 callersFunctionread_results
(filename, data_type: str, is_gt=False, is_ignore=False)
utils/io.py:36
↓ 3 callersFunctionunzip_objs
(objs)
utils/io.py:105
↓ 3 callersMethodupdate_features
(self, feat)
tracker/multitracker.py:29
↓ 2 callersFunctionbbox_iou
Returns the IoU of two bounding boxes
utils/utils.py:199
↓ 2 callersFunctioneval_seq
Processes the video sequence given and provides the output of tracking result (write the results in video file) It uses JDE model f
track.py:41
↓ 2 callersFunctiongenerate_anchor
(nGh, nGw, anchor_wh)
utils/utils.py:373
↓ 2 callersFunctionget_color
(idx)
utils/visualization.py:14
↓ 2 callersMethodget_data
(self, img_path, label_path)
utils/datasets.py:152
↓ 2 callersMethodmark_removed
(self)
tracker/basetrack.py:51
↓ 2 callersMethodnext_id
()
tracker/basetrack.py:35
↓ 2 callersMethodproject
Project state distribution to measurement space. Parameters ---------- mean : ndarray The state's mean vecto
utils/kalman_filter.py:125
↓ 2 callersMethodre_activate
(self, new_track, frame_id, new_id=False)
tracker/multitracker.py:71
↓ 2 callersMethodreset_accumulator
(self)
utils/evaluation.py:26
↓ 2 callersFunctionscale_coords
(img_size, coords, img0_shape)
utils/utils.py:103
↓ 2 callersFunctionsub_stracks
(tlista, tlistb)
tracker/multitracker.py:364
↓ 2 callersFunctionxywh2xyxy
(x)
utils/utils.py:91
↓ 1 callersMethodactivate
Start a new tracklet
tracker/multitracker.py:59
↓ 1 callersFunctionap_per_class
Computes the average precision, given the recall and precision curves. Method originally from https://github.com/rafaelpadilla/Object-Detection-
utils/utils.py:117
↓ 1 callersFunctionbuild_targets_max
returns nT, nCorrect, tx, ty, tw, th, tconf, tcls
utils/utils.py:230
↓ 1 callersFunctionbuild_targets_thres
(target, anchor_wh, nA, nC, nGh, nGw)
utils/utils.py:315
↓ 1 callersFunctioncompute_ap
Computes the average precision, given the recall and precision curves. Code originally from https://github.com/rbgirshick/py-faster-rcnn. #
utils/utils.py:171
↓ 1 callersFunctioncreate_grids
(self, img_size, nGh, nGw)
models.py:292
↓ 1 callersFunctioncreate_modules
Constructs module list of layer blocks from module configuration in module_defs
models.py:17
↓ 1 callersFunctiondecode_delta
(delta, fg_anchor_list)
utils/utils.py:395
↓ 1 callersFunctiondecode_delta_map
:param: delta_map, shape (nB, nA, nGh, nGw, 4) :param: anchors, shape (nA,4)
utils/utils.py:405
↓ 1 callersFunctionencode_delta
(gt_box_list, fg_anchor_list)
utils/utils.py:384
↓ 1 callersMethodeval_file
(self, filename)
utils/evaluation.py:69
↓ 1 callersMethodeval_frame
(self, frame_id, trk_tlwhs, trk_ids, rtn_events=False)
utils/evaluation.py:29
↓ 1 callersFunctionextract_ped_per_frame
( cfg, input_root, output_root, weights, batch_size=16,
extract_ped_per_frame.py:14
↓ 1 callersFunctionfast_nms
Vectorized, approximated, fast NMS, adopted from YOLACT: https://github.com/dbolya/yolact/blob/master/layers/functions/detection.py Th
utils/utils.py:473
↓ 1 callersMethodgating_distance
Compute gating distance between state distribution and measurements. A suitable distance threshold can be obtained from `chi2inv95`. If
utils/kalman_filter.py:230
↓ 1 callersFunctionget_logger
(name='root')
utils/log.py:4
↓ 1 callersMethodget_size
(self, vw, vh, dw, dh)
utils/datasets.py:97
↓ 1 callersMethodget_summary
(accs, names, metrics=('mota', 'num_switches', 'idp', 'idr', 'idf1', 'precision', 'recall'))
utils/evaluation.py:82
↓ 1 callersFunctioninit_seeds
(seed=0)
utils/utils.py:22
↓ 1 callersMethodinitiate
Create track from unassociated measurement. Parameters ---------- measurement : ndarray Bounding box coordin
utils/kalman_filter.py:55
↓ 1 callersFunctionintersect
We resize both tensors to [A,B,2] without new malloc: [A,2] -> [A,1,2] -> [A,B,2] [B,2] -> [1,B,2] -> [A,B,2] Then we compute the area
utils/utils.py:501
↓ 1 callersFunctionious
Compute cost based on IoU :type atlbrs: list[tlbr] | np.ndarray :type atlbrs: list[tlbr] | np.ndarray :rtype ious np.ndarray
tracker/matching.py:40
↓ 1 callersFunctionjaccard
Compute the jaccard overlap of two sets of boxes. The jaccard overlap is simply the intersection over union of two boxes. Here we operate on
utils/utils.py:524
↓ 1 callersMethodload_annotations
(self)
utils/evaluation.py:19
↓ 1 callersFunctionmain
(opt, data_root='/data/MOT16/train', det_root=None, seqs=('MOT16-05',), exp_name='demo', save_image
track.py:116
↓ 1 callersMethodmark_lost
(self)
tracker/basetrack.py:48
↓ 1 callersMethodmulti_predict
(stracks, kalman_filter)
tracker/multitracker.py:46
↓ 1 callersFunctionrandom_affine
(img, targets=None, degrees=(-10, 10), translate=(.1, .1), scale=(.9, 1.1), shear=(-2, 2),
utils/datasets.py:252
↓ 1 callersFunctionremove_duplicate_stracks
(stracksa, stracksb)
tracker/multitracker.py:374
↓ 1 callersFunctionreturn_torch_unique_index
(u, uv)
utils/utils.py:553
↓ 1 callersMethodsave_summary
(summary, filename)
utils/evaluation.py:99
↓ 1 callersFunctiontest
( cfg, data_cfg, weights, batch_size=16, iou_thres=0.5,
test.py:14
↓ 1 callersFunctiontest_emb
( cfg, data_cfg, weights, batch_size=16, iou_
test.py:147
↓ 1 callersMethodtic
(self)
utils/timer.py:22
↓ 1 callersMethodtlbr_to_tlwh
(tlbr)
tracker/multitracker.py:144
↓ 1 callersMethodto_xyah
(self)
tracker/multitracker.py:139
↓ 1 callersMethodtoc
(self, average=True)
utils/timer.py:27
↓ 1 callersFunctiontrack
(opt)
demo.py:39
↓ 1 callersFunctiontrain
( cfg, data_cfg, weights_from="", weights_to="", save_every=10,
train.py:14
↓ 1 callersMethodupdate
Run Kalman filter correction step. Parameters ---------- mean : ndarray The predicted state's mean vector (8
utils/kalman_filter.py:196
↓ 1 callersMethodupdate
Processes the image frame and finds bounding box(detections). Associates the detection with corresponding tracklets and also hand
tracker/multitracker.py:179
↓ 1 callersFunctionwrite_results
(filename, results, data_type)
track.py:19
↓ 1 callersFunctionxyxy2xywh
(x)
utils/utils.py:79
Method__getitem__
(self, idx)
utils/datasets.py:57
Method__getitem__
(self, files_index)
utils/datasets.py:147
Method__getitem__
Iterator function for train dataset
utils/datasets.py:398
Method__init__
(self)
models.py:95
Method__init__
(self, scale_factor=1, mode='nearest')
models.py:105
Method__init__
(self, anchors, nC, nID, nE, img_size, yolo_layer)
models.py:115
Method__init__
(self, data_root, seq_name, data_type)
utils/evaluation.py:11
Method__init__
(self)
utils/kalman_filter.py:40
Method__init__
(self, path, img_size=(1088, 608))
utils/datasets.py:17
Method__init__
(self, path, img_size=(1088, 608))
utils/datasets.py:80
Method__init__
(self, path, img_size=(1088,608), augment=False, transforms=None)
utils/datasets.py:131
Method__init__
(self, root, paths, img_size=(1088,608), augment=False, transforms=None)
utils/datasets.py:345
Method__init__
(self)
utils/timer.py:13
Method__init__
(self, tlwh, score, temp_feat, buffer_size=30)
tracker/multitracker.py:13
Method__init__
(self, opt, frame_rate=30)
tracker/multitracker.py:161
Method__iter__
(self)
utils/datasets.py:32
Method__iter__
(self)
utils/datasets.py:102
Method__len__
(self)
utils/datasets.py:75
Method__len__
(self)
utils/datasets.py:126
Method__len__
(self)
utils/datasets.py:235
Method__next__
(self)
utils/datasets.py:36
Method__next__
(self)
utils/datasets.py:106
Method__repr__
(self)
tracker/multitracker.py:156
Methodactivate
(self, *args)
tracker/basetrack.py:39
Methodclear
(self)
utils/timer.py:38
Functioncollate_fn
(batch)
utils/datasets.py:326
Functionembedding_distance
:param tracks: list[STrack] :param detections: list[BaseTrack] :param metric: :return: cost_matrix np.ndarray
tracker/matching.py:80
Methodend_frame
(self)
tracker/basetrack.py:31
Functionfloat3
(x)
utils/utils.py:18
Methodforward
(self, x)
models.py:98
Methodforward
(self, x)
models.py:110
Methodforward
(self, p_cat, img_size, targets=None, classifier=None, test_emb=False)
models.py:139
next →1–100 of 130, ranked by callers