Spaces:
Running
Running
File size: 910 Bytes
67a9b5d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 |
import numpy as np
__all__ = ['scale_2d_points']
def scale_2d_points(points, x_scale=1, y_scale=1, x_center=0, y_center=0, copy=True):
"""Scale 2d points.
Args:
points: (..., 2N)
x_scale: scale factor in x dimension
y_scale: scale factor in y dimension
x_center: scale center in x dimension
y_center: scale center in y dimension
"""
points = np.array(points, dtype=np.float32, copy=copy)
x_scale = np.asarray(x_scale, np.float32)
y_scale = np.asarray(y_scale, np.float32)
x_center = np.asarray(x_center, np.float32)
y_center = np.asarray(y_center, np.float32)
x_shift = 1 - x_scale
y_shift = 1 - y_scale
x_shift *= x_center
y_shift *= y_center
points[..., 0::2] *= x_scale
points[..., 1::2] *= y_scale
points[..., 0::2] += x_shift
points[..., 1::2] += y_shift
return points
|