MCPcopy Create free account

hub / github.com/Sharpiless/Yolov5-deepsort-inference / functions

Functions240 in github.com/Sharpiless/Yolov5-deepsort-inference

↓ 7 callersMethodinfo
(self, verbose=False, img_size=640)
models/yolo.py:180
↓ 7 callersMethodplot
(self, save_dir='', names=())
utils/metrics.py:159
↓ 6 callersMethod__init__
(self, c1, c2, k=(5, 9, 13))
models/common.py:71
↓ 6 callersMethod__init__
(self, c1, c2, n=1, shortcut=True, g=1, e=0.5)
models/experimental.py:27
↓ 5 callersMethodis_confirmed
Returns True if this track is confirmed.
deep_sort/deep_sort/sort/track.py:160
↓ 4 callersFunctionmake_layers
(c_in,c_out,repeat_times, is_downsample=False)
deep_sort/deep_sort/deep/model.py:39
↓ 4 callersFunctionmetric
(k)
utils/autoanchor.py:29
↓ 4 callersMethodupdate
Perform Kalman filter measurement update step and update the feature cache. Parameters ---------- kf : kalman_filter.
deep_sort/deep_sort/sort/track.py:126
↓ 3 callersMethodframe_exists
Args: frame_id (int): Returns: bool: true if frame_id is recognized
deep_sort/utils/json_logger.py:185
↓ 3 callersFunctionmake_layers
(c_in,c_out,repeat_times, is_downsample=False)
deep_sort/deep_sort/deep/original_model.py:39
↓ 3 callersFunctionprint_results
(k)
utils/autoanchor.py:84
↓ 3 callersFunctionread_results
(filename, data_type: str, is_gt=False, is_ignore=False)
deep_sort/utils/io.py:57
↓ 3 callersMethodto_tlwh
Get current position in bounding box format `(top left x, top left y, width, height)`. Returns ------- ndarray
deep_sort/deep_sort/sort/track.py:83
↓ 3 callersMethodto_xyah
Convert bounding box to format `(center x, center y, aspect ratio, height)`, where the aspect ratio is `width / height`.
deep_sort/deep_sort/sort/detection.py:42
↓ 3 callersFunctionunzip_objs
(objs)
deep_sort/utils/io.py:126
↓ 2 callersFunctionDWConv
(c1, c2, k=1, s=1, act=True)
models/common.py:17
↓ 2 callersMethodadd_label
(self, category, confidence)
deep_sort/utils/json_logger.py:85
↓ 2 callersFunctionanchor_fitness
(k)
utils/autoanchor.py:80
↓ 2 callersMethodbbox_exists
Args: frame_id: bbox_id: Returns: bool: if bbox exists in frame bboxes list
deep_sort/utils/json_logger.py:213
↓ 2 callersFunctionbox_area
(box)
utils/general.py:248
↓ 2 callersFunctioncheck_anchor_order
(m)
utils/autoanchor.py:10
↓ 2 callersFunctioncompute_color_for_labels
Simple function that adds fixed color depending on the class
deep_sort/utils/draw.py:7
↓ 2 callersMethoddic
(self)
deep_sort/utils/json_logger.py:26
↓ 2 callersMethodforward_once
(self, x, profile=False)
models/yolo.py:114
↓ 2 callersMethodfuse
(self)
models/yolo.py:156
↓ 2 callersFunctionis_parallel
(model)
utils/torch_utils.py:73
↓ 2 callersMethodjson_output
Args: output_name: Returns: None Notes: It creates the json output with `output_name` n
deep_sort/utils/json_logger.py:306
↓ 2 callersFunctionmake_divisible
(x, divisor)
utils/general.py:97
↓ 2 callersMethodmerge_from_file
(self, config_file)
deep_sort/utils/parser.py:21
↓ 2 callersFunctionnon_max_suppression
Performs Non-Maximum Suppression (NMS) on inference results Returns: detections with shape: nx6 (x1, y1, x2, y2, conf, cls)
utils/general.py:271
↓ 2 callersMethodproject
Project state distribution to measurement space. Parameters ---------- mean : ndarray The state's mean vector (8
deep_sort/deep_sort/sort/kalman_filter.py:125
↓ 2 callersMethodreset_accumulator
(self)
deep_sort/utils/evaluation.py:26
↓ 2 callersFunctionscale_coords
(img1_shape, coords, img0_shape, ratio_pad=None)
utils/general.py:162
↓ 2 callersFunctiontime_synchronized
()
utils/torch_utils.py:68
↓ 2 callersFunctionxywh2xyxy
(x)
utils/general.py:152
↓ 1 callersMethod__init__
(self, num_classes=625 ,reid=False)
deep_sort/deep_sort/deep/original_model.py:49
↓ 1 callersMethod__init__
(self, num_classes=751 ,reid=False)
deep_sort/deep_sort/deep/model.py:49
↓ 1 callersMethod__init__
(self, cfg='yolov5m.yaml', ch=3, nc=None)
models/yolo.py:63
↓ 1 callersFunction_cosine_distance
Compute pair-wise cosine distance between points in `a` and `b`. Parameters ---------- a : array_like An NxM matrix of N samples
deep_sort/deep_sort/sort/nn_matching.py:31
↓ 1 callersMethod_get_features
(self, bbox_xywh, ori_img)
deep_sort/deep_sort/deep_sort.py:103
↓ 1 callersMethod_initialize_biases
(self, cf=None)
models/yolo.py:135
↓ 1 callersMethod_initiate_track
(self, detection)
deep_sort/deep_sort/sort/tracker.py:133
↓ 1 callersMethod_make_grid
(nx=20, ny=20)
models/yolo.py:57
↓ 1 callersMethod_match
(self, detections)
deep_sort/deep_sort/sort/tracker.py:93
↓ 1 callersFunction_pdist
Compute pair-wise squared distance between points in `a` and `b`. Parameters ---------- a : array_like An NxM matrix of N samples
deep_sort/deep_sort/sort/nn_matching.py:5
↓ 1 callersMethod_preprocess
TODO: 1. to float with scale from 0 to 1 2. resize to (64, 128) as Market1501 dataset did 3. concatenate
deep_sort/deep_sort/deep/feature_extractor.py:26
↓ 1 callersMethod_tlwh_to_xyxy
TODO: Convert bbox from xtl_ytl_w_h to xc_yc_w_h Thanks JieChen91@github.com for reporting this bug!
deep_sort/deep_sort/deep_sort.py:81
↓ 1 callersMethod_xywh_to_tlwh
(bbox_xywh)
deep_sort/deep_sort/deep_sort.py:63
↓ 1 callersMethod_xywh_to_xyxy
(self, bbox_xywh)
deep_sort/deep_sort/deep_sort.py:73
↓ 1 callersMethodadd_bbox
(self, bbox_id: int, top: int, left: int, width: int, height: int)
deep_sort/utils/json_logger.py:115
↓ 1 callersFunctionassert_in
(file, files_to_check)
deep_sort/utils/asserts.py:4
↓ 1 callersFunctionattempt_download
(weights)
utils/google_utils.py:18
↓ 1 callersFunctionattempt_load
(weights, map_location=None)
models/experimental.py:132
↓ 1 callersFunctionautopad
(k, p=None)
models/common.py:10
↓ 1 callersMethodbackward
(ctx, grad_output)
utils/activations.py:54
↓ 1 callersFunctionbox_iou
Return intersection-over-union (Jaccard index) of boxes. Both sets of boxes are expected to be in (x1, y1, x2, y2) format. Arguments:
utils/general.py:235
↓ 1 callersMethodbuild_config
(self)
utils/BaseDetector.py:13
↓ 1 callersFunctionclip_coords
(boxes, img_shape)
utils/general.py:180
↓ 1 callersFunctioncompute_ap
Compute the average precision, given the recall and precision curves. Source: https://github.com/rbgirshick/py-faster-rcnn. # Arguments
utils/metrics.py:79
↓ 1 callersFunctioncopy_attr
(a, b, include=(), exclude=())
utils/torch_utils.py:199
↓ 1 callersMethoddetect
(self)
utils/BaseDetector.py:49
↓ 1 callersMethoddistance
Compute distance between features and targets. Parameters ---------- features : ndarray An NxM matrix of N featur
deep_sort/deep_sort/sort/nn_matching.py:156
↓ 1 callersFunctiondraw_curve
(epoch, train_loss, train_err, test_loss, test_err)
deep_sort/deep_sort/deep/train.py:154
↓ 1 callersMethodeval_frame
(self, frame_id, trk_tlwhs, trk_ids, rtn_events=False)
deep_sort/utils/evaluation.py:29
↓ 1 callersMethodfeedCap
(self, im, func_status)
utils/BaseDetector.py:25
↓ 1 callersMethodfind_bbox
Args: frame_id: bbox_id: Returns: bbox_id (int): Raises: ValueError: if bb
deep_sort/utils/json_logger.py:227
↓ 1 callersFunctionfitness
(x)
utils/metrics.py:12
↓ 1 callersMethodforward
(self, x, augment=False, profile=False)
models/yolo.py:94
↓ 1 callersFunctionfuse_conv_and_bn
(conv, bn)
utils/torch_utils.py:119
↓ 1 callersMethodgating_distance
Compute gating distance between state distribution and measurements. A suitable distance threshold can be obtained from `chi2inv95`. If
deep_sort/deep_sort/sort/kalman_filter.py:188
↓ 1 callersFunctionget_config
(config_file=None)
deep_sort/utils/parser.py:30
↓ 1 callersFunctionget_token
(cookie="./cookie")
utils/google_utils.py:90
↓ 1 callersFunctiongsutil_getsize
(url='')
utils/google_utils.py:12
↓ 1 callersMethodinit_model
(self)
AIDetector_pytorch.py:16
↓ 1 callersFunctioninit_torch_seeds
(seed=0)
utils/torch_utils.py:31
↓ 1 callersFunctioninitialize_weights
(model)
utils/torch_utils.py:82
↓ 1 callersMethodinitiate
Create track from unassociated measurement. Parameters ---------- measurement : ndarray Bounding box coordinates
deep_sort/deep_sort/sort/kalman_filter.py:55
↓ 1 callersFunctioniou
Computer intersection over union. Parameters ---------- bbox : ndarray A bounding box in format `(top left x, top left y, width,
deep_sort/deep_sort/sort/iou_matching.py:7
↓ 1 callersMethodis_deleted
Returns True if this track is dead and should be deleted.
deep_sort/deep_sort/sort/track.py:164
↓ 1 callersFunctionkmean_anchors
Creates kmeans-evolved anchors from training dataset Arguments: path: path to dataset *.yaml, or a loaded dataset n:
utils/autoanchor.py:55
↓ 1 callersMethodlabels_full
(self, value)
deep_sort/utils/json_logger.py:89
↓ 1 callersFunctionletterbox
(img, new_shape=(640, 640), color=(114, 114, 114), auto=True, scaleFill=False, scaleup=True)
utils/general.py:469
↓ 1 callersMethodlist
(values)
deep_sort/utils/json_logger.py:39
↓ 1 callersMethodload_annotations
(self)
deep_sort/utils/evaluation.py:19
↓ 1 callersFunctionlr_decay
()
deep_sort/deep_sort/deep/train.py:172
↓ 1 callersFunctionmain
()
demo.py:5
↓ 1 callersFunctionmain
()
deep_sort/deep_sort/deep/train.py:179
↓ 1 callersMethodmark_missed
Mark this track as missed (no association at the current time step).
deep_sort/deep_sort/sort/track.py:147
↓ 1 callersFunctionmin_cost_matching
Solve linear assignment problem. Parameters ---------- distance_metric : Callable[List[Track], List[Detection], List[int], List[int]) ->
deep_sort/deep_sort/sort/linear_assignment.py:12
↓ 1 callersFunctionmodel_info
(model, verbose=False, img_size=640)
utils/torch_utils.py:142
↓ 1 callersMethodnms
(self, mode=True)
models/yolo.py:166
↓ 1 callersFunctionnon_max_suppression
Suppress overlapping detections. Original code from [1]_ has been adapted to include confidence score. .. [1] http://www.pyimagesearch.com/2
deep_sort/deep_sort/sort/preprocessing.py:6
↓ 1 callersMethodoutput
(self)
deep_sort/utils/json_logger.py:300
↓ 1 callersFunctionparse_model
(d, ch)
models/yolo.py:184
↓ 1 callersMethodpartial_fit
Update the distance metric with new data. Parameters ---------- features : ndarray An NxM matrix of N features of
deep_sort/deep_sort/sort/nn_matching.py:137
↓ 1 callersFunctionplot_bboxes
(image, bboxes, line_thickness=None)
tracker.py:16
↓ 1 callersFunctionplot_pr_curve
(px, py, ap, save_dir='.', names=())
utils/metrics.py:185
↓ 1 callersMethodpredict
Propagate the state distribution to the current time step using a Kalman filter prediction step. Parameters ----------
deep_sort/deep_sort/sort/track.py:112
↓ 1 callersMethodpreprocess
(self, img)
AIDetector_pytorch.py:29
↓ 1 callersFunctionscale_img
(img, ratio=1.0, same_shape=False)
utils/torch_utils.py:185
next →1–100 of 240, ranked by callers