xiaoxue_admin
/
supervision
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix(pre_commit): 🎨 auto format pre-commit hooks

This commit is contained in:
pre-commit-ci[bot] 2024-04-04 04:59:29 +00:00
parent 0cf31d7e55
commit 8de73738c7
2 changed files with 22 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
supervision/geometry/utils.py
View File

@ -36,7 +36,7 @@ def get_polygon_center(polygon: np.ndarray) -> Point:
shifted_polygon = np.roll(polygon, 1, axis=0)
points = (shifted_polygon + polygon) / 2
vectors = shifted_polygon - polygon
mass = np.sum(vectors ** 2, axis=1) ** 0.5
mass = np.sum(vectors**2, axis=1) ** 0.5
center = ((mass @ points) / np.sum(mass)).round()
return Point(x=center[0], y=center[1])

40
test/geometry/test_utils.py
View File

@ -7,32 +7,32 @@ from supervision.geometry.utils import get_polygon_center
def generate_test_polygon(n: int) -> np.ndarray:
"""
Generate a semicircle with a given number of points.
Parameters:
n (int): amount of points in polygon
Generate a semicircle with a given number of points.
Returns:
Polygon: test polygon in the form of a semicircle.
Examples:
```python
from supervision.geometry.utils import get_polygon_center
import numpy as np
test_polygon = generate_test_data(1000)
get_polygon_center(test_polygon)
Point(x=500, y=1212)
```
Parameters:
n (int): amount of points in polygon
Returns:
Polygon: test polygon in the form of a semicircle.
Examples:
```python
from supervision.geometry.utils import get_polygon_center
import numpy as np
test_polygon = generate_test_data(1000)
get_polygon_center(test_polygon)
Point(x=500, y=1212)
```
"""
r: int = n // 2
x_axis = np.linspace(0, 2 * r, n)
y_axis = (r ** 2 - (x_axis - r) ** 2) ** 0.5 + 2 * r
y_axis = (r**2 - (x_axis - r) ** 2) ** 0.5 + 2 * r
polygon = np.array([x_axis, y_axis]).T
return polygon
@pytest.mark.parametrize(
"polygon, expected_result",