HongChuang/ultralytics-ascend
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 3

ultralytics 8.0.222 introduce new ultralytics.solutions (#6759)

Browse source 
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: Yujiao Jiang <36917338+Alex-JYJ@users.noreply.github.com>
Co-authored-by: snyk-bot <snyk-bot@snyk.io>
This commit is contained in:
Glenn Jocher 2023-12-03 14:59:33 +01:00 • committed by GitHub
parent 63fcc2a9ff
commit d12411ec0d
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
6 changed files with 22 additions and 22 deletions
Download patch file Download diff file Expand all files Collapse all files

6
ultralytics/trackers/utils/matching.py
View file

@ -67,7 +67,7 @@ def iou_distance(atracks: list, btracks: list) -> np.ndarray:
btracks (list[STrack] | list[np.ndarray]): List of tracks 'b' or bounding boxes.
Returns:
np.ndarray: Cost matrix computed based on IoU.
(np.ndarray): Cost matrix computed based on IoU.
"""
if (len(atracks) > 0 and isinstance(atracks[0], np.ndarray)) \
@ -96,7 +96,7 @@ def embedding_distance(tracks: list, detections: list, metric: str = 'cosine') -
metric (str, optional): Metric for distance computation. Defaults to 'cosine'.
Returns:
np.ndarray: Cost matrix computed based on embeddings.
(np.ndarray): Cost matrix computed based on embeddings.
"""
cost_matrix = np.zeros((len(tracks), len(detections)), dtype=np.float32)
@ -119,7 +119,7 @@ def fuse_score(cost_matrix: np.ndarray, detections: list) -> np.ndarray:
detections (list[BaseTrack]): List of detections with scores.
Returns:
np.ndarray: Fused similarity matrix.
(np.ndarray): Fused similarity matrix.
"""
if cost_matrix.size == 0: