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

Expand Model method type hinting (#8279)

Browse source 
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
This commit is contained in:
Glenn Jocher 2024-02-19 14:24:30 +01:00 • committed by GitHub
parent fbed8499da
commit 42744a1717
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
6 changed files with 104 additions and 50 deletions
Download patch file Download diff file Expand all files Collapse all files

4
ultralytics/data/converter.py
View file

@ -418,8 +418,8 @@ def min_index(arr1, arr2):
Find a pair of indexes with the shortest distance between two arrays of 2D points.
Args:
arr1 (np.array): A NumPy array of shape (N, 2) representing N 2D points.
arr2 (np.array): A NumPy array of shape (M, 2) representing M 2D points.
arr1 (np.ndarray): A NumPy array of shape (N, 2) representing N 2D points.
arr2 (np.ndarray): A NumPy array of shape (M, 2) representing M 2D points.
Returns:
(tuple): A tuple containing the indexes of the points with the shortest distance in arr1 and arr2 respectively.