MCPcopy Create free account

hub / github.com/TsMask/deep-sort-yolov4 / functions

Functions82 in github.com/TsMask/deep-sort-yolov4

↓ 41 callersFunctionDarknetConv2D_BN_Leaky
Darknet Convolution2D followed by BatchNormalization and LeakyReLU.
yolo4/model.py:53
↓ 8 callersFunctionDarknetConv2D_BN_Mish
Darknet Convolution2D followed by BatchNormalization and LeakyReLU.
yolo4/model.py:62
↓ 7 callersFunctioncompose
Compose arbitrarily many functions, evaluated left to right. Reference: https://mathieularose.com/function-composition-in-python/
yolo4/model.py:753
↓ 6 callersFunctionDarknetConv2D
Wrapper to set Darknet parameters for Convolution2D.
yolo4/model.py:46
↓ 5 callersMethodis_confirmed
Returns True if this track is confirmed.
deep_sort/track.py:160
↓ 5 callersFunctionresblock_body
A series of resblocks starting with a downsampling Convolution2D
yolo4/model.py:71
↓ 4 callersMethodupdate
Perform Kalman filter measurement update step and update the feature cache. Parameters ---------- kf : kalman_filter.
deep_sort/track.py:126
↓ 3 callersMethodto_xyah
Convert bounding box to format `(center x, center y, aspect ratio, height)`, where the aspect ratio is `width / height`.
deep_sort/detection.py:44
↓ 3 callersFunctionyolo_head
Convert final layer features to bounding box parameters.
yolo4/model.py:180
↓ 2 callersFunctionbox_iou
Return iou tensor Parameters ---------- b1: tensor, shape=(i1,...,iN, 4), xywh b2: tensor, shape=(j, 4), xywh Returns -
yolo4/model.py:420
↓ 2 callersFunctionencoder
(image, boxes)
deep_sort/generate_detections.py:104
↓ 2 callersMethodproject
Project state distribution to measurement space. Parameters ---------- mean : ndarray The state's mean vector (8
deep_sort/kalman_filter.py:125
↓ 2 callersFunctionsigmoid_focal_loss
Compute sigmoid focal loss. Reference Paper: "Focal Loss for Dense Object Detection" https://arxiv.org/abs/1708.02002 #
yolo4/model.py:390
↓ 2 callersMethodto_tlbr
Get current position in bounding box format `(min x, miny, max x, max y)`. Returns ------- ndarray The bo
deep_sort/track.py:98
↓ 2 callersMethodto_tlwh
Get current position in bounding box format `(top left x, top left y, width, height)`. Returns ------- ndarray
deep_sort/track.py:83
↓ 2 callersFunctionyolo_eval
Evaluate YOLO model on given input and return filtered boxes.
yolo4/model.py:242
↓ 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/nn_matching.py:31
↓ 1 callersMethod_generate
(self)
yolo4/yolo.py:71
↓ 1 callersMethod_initiate_track
(self, detection)
deep_sort/tracker.py:133
↓ 1 callersMethod_letterbox_image
resize image with unchanged aspect ratio using padding
yolo4/yolo.py:57
↓ 1 callersMethod_match
(self, detections)
deep_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/nn_matching.py:5
↓ 1 callersMethod_random_colors
(self,N)
yolo4/yolo.py:47
↓ 1 callersFunction_run_in_batches
(f, data_dict, out, batch_size)
deep_sort/generate_detections.py:11
↓ 1 callersFunction_smooth_labels
(y_true, label_smoothing)
yolo4/model.py:565
↓ 1 callersFunctionbox_diou
Calculate DIoU loss on anchor boxes Reference Paper: "Distance-IoU Loss: Faster and Better Learning for Bounding Box Regression"
yolo4/model.py:508
↓ 1 callersFunctionbox_giou
Calculate GIoU loss on anchor boxes Reference Paper: "Generalized Intersection over Union: A Metric and A Loss for Bounding Box Regre
yolo4/model.py:459
↓ 1 callersMethodclose_session
(self)
convertToH5.py:120
↓ 1 callersFunctioncreate_box_encoder
(model_filename, input_name="images", output_name="features", batch_size=32)
deep_sort/generate_detections.py:99
↓ 1 callersFunctiondarknet_body
Darknent body having 52 Convolution2D layers
yolo4/model.py:85
↓ 1 callersMethoddetect_image
(self, image)
yolo4/yolo.py:90
↓ 1 callersMethoddistance
Compute distance between features and targets. Parameters ---------- features : ndarray An NxM matrix of N featur
deep_sort/nn_matching.py:156
↓ 1 callersFunctionextract_image_patch
Extract image patch from bounding box. Parameters ---------- image : ndarray The full image. bbox : array_like The bo
deep_sort/generate_detections.py:25
↓ 1 callersMethodgating_distance
Compute gating distance between state distribution and measurements. A suitable distance threshold can be obtained from `chi2inv95`. If
deep_sort/kalman_filter.py:188
↓ 1 callersFunctiongenerate_detections
Generate detections with features. Parameters ---------- encoder : Callable[image, ndarray] -> ndarray The encoder function takes
deep_sort/generate_detections.py:119
↓ 1 callersMethodinitiate
Create track from unassociated measurement. Parameters ---------- measurement : ndarray Bounding box coordinates
deep_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/iou_matching.py:7
↓ 1 callersMethodis_deleted
Returns True if this track is dead and should be deleted.
deep_sort/track.py:164
↓ 1 callersMethodload_weights
(self,model, weights_file)
convertToH5.py:49
↓ 1 callersMethodload_yolo
(self)
convertToH5.py:90
↓ 1 callersFunctionmain
()
deep_sort/generate_detections.py:206
↓ 1 callersMethodmark_missed
Mark this track as missed (no association at the current time step).
deep_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/linear_assignment.py:13
↓ 1 callersFunctionparse_args
Parse command line arguments.
deep_sort/generate_detections.py:185
↓ 1 callersMethodpartial_fit
Update the distance metric with new data. Parameters ---------- features : ndarray An NxM matrix of N features of
deep_sort/nn_matching.py:137
↓ 1 callersMethodpredict
Propagate the state distribution to the current time step using a Kalman filter prediction step. Parameters ----------
deep_sort/track.py:112
↓ 1 callersFunctionsoftmax_focal_loss
Compute softmax focal loss. Reference Paper: "Focal Loss for Dense Object Detection" https://arxiv.org/abs/1708.02002 #
yolo4/model.py:354
↓ 1 callersFunctionyolo4_body
Create YOLO_V4 model CNN body in Keras.
yolo4/model.py:108
↓ 1 callersFunctionyolo_boxes_and_scores
Process Conv layer output
yolo4/model.py:232
↓ 1 callersFunctionyolo_correct_boxes
Get corrected boxes
yolo4/model.py:207
Method__call__
(self, data_x, batch_size=32)
deep_sort/generate_detections.py:91
Method__init__
(self, score, iou, model_path, weights_path,input_size, gpu_num=1)
convertToH5.py:39
Method__init__
(self,model_path, score)
yolo4/yolo.py:34
Method__init__
(self, **kwargs)
yolo4/model.py:30
Method__init__
(self, metric, matching_threshold, budget=None)
deep_sort/nn_matching.py:123
Method__init__
(self, tlwh, score, classe, color, feature)
deep_sort/detection.py:29
Method__init__
(self, checkpoint_filename, input_name="images", output_name="features")
deep_sort/generate_detections.py:74
Method__init__
(self)
deep_sort/kalman_filter.py:40
Method__init__
(self, mean, covariance, track_id, n_init, max_age, feature=None)
deep_sort/track.py:66
Method__init__
(self, metric, max_iou_distance=0.7, max_age=30, n_init=3)
deep_sort/tracker.py:40
Function_nn_cosine_distance
Helper function for nearest neighbor distance metric (cosine). Parameters ---------- x : ndarray A matrix of N row-vectors (samp
deep_sort/nn_matching.py:78
Function_nn_euclidean_distance
Helper function for nearest neighbor distance metric (Euclidean). Parameters ---------- x : ndarray A matrix of N row-vectors (s
deep_sort/nn_matching.py:57
Methodcall
(self, inputs)
yolo4/model.py:34
Methodclose_session
(self)
yolo4/yolo.py:121
Methodcompute_output_shape
(self, input_shape)
yolo4/model.py:41
Functiongate_cost_matrix
Invalidate infeasible entries in cost matrix based on the state distributions obtained by Kalman filtering. Parameters ---------- kf
deep_sort/linear_assignment.py:148
Methodgated_metric
(tracks, dets, track_indices, detection_indices)
deep_sort/tracker.py:95
Methodget_config
(self)
yolo4/model.py:37
Functioniou_cost
An intersection over union distance metric. Parameters ---------- tracks : List[deep_sort.track.Track] A list of tracks. dete
deep_sort/iou_matching.py:42
Methodis_tentative
Returns True if this track is tentative (unconfirmed).
deep_sort/track.py:155
Functionloop_body
(b, ignore_mask)
yolo4/model.py:617
Functionmake_last_layers
6 Conv2D_BN_Leaky layers followed by a Conv2D_linear layer
yolo4/model.py:95
Functionmatching_cascade
Run matching cascade. Parameters ---------- distance_metric : Callable[List[Track], List[Detection], List[int], List[int]) -> ndarray
deep_sort/linear_assignment.py:82
Functionnon_max_suppression
Suppress overlapping detections. Original code from [1]_ has been adapted to include confidence score. .. [1] http://www.pyimagesearch.com/2
deep_sort/preprocessing.py:6
Methodpredict
Run Kalman filter prediction step. Parameters ---------- mean : ndarray The 8 dimensional mean vector of the obje
deep_sort/kalman_filter.py:88
Methodpredict
Propagate track state distributions one time step forward. This function should be called once every time step, before `update`.
deep_sort/tracker.py:50
Functionpreprocess_true_boxes
Preprocess true boxes to training input format Parameters ---------- true_boxes: array, shape=(m, T, 5) Absolute x_min, y_min, x_
yolo4/model.py:283
Methodto_tlbr
Convert bounding box to format `(min x, min y, max x, max y)`, i.e., `(top left, bottom right)`.
deep_sort/detection.py:36
Methodupdate
Run Kalman filter correction step. Parameters ---------- mean : ndarray The predicted state's mean vector (8 dime
deep_sort/kalman_filter.py:154
Methodupdate
Perform measurement update and track management. Parameters ---------- detections : List[deep_sort.detection.Detection]
deep_sort/tracker.py:58
Functionyolo4_loss
Return yolo4_loss tensor Parameters ---------- yolo_outputs: list of tensor, the output of yolo_body or tiny_yolo_body y_true: list o
yolo4/model.py:569
Functionyolo_loss
Return yolo_loss tensor Parameters ---------- yolo_outputs: list of tensor, the output of yolo_body or tiny_yolo_body y_true: list of
yolo4/model.py:684