首页 > 解决方案 > 用 numpy 计算坐标的新位置

问题描述

我有一个用于关键点检测的图像数据集。每个图像都用一个关键点 (x|y) 标记。

我使用 numpy 翻转图像以进行数据增强。

我使用以下代码翻转图像水平:

img = img[:, ::-1]

与此代码垂直

 img = img[::-1]

到目前为止,一切都很好。但我还必须重新计算关键点(标签)([85 35])我知道它的基本数学,但我还没有找到解决方案。

提前致谢。

标签: pythonnumpydata-augmentation

解决方案


使用旋转矩阵:

x_new = x_old * np.cos(alpha) - y_old * np.sin(alpha)
y_new = x_old * np.sin(alpha) + y_old * np.cos(alpha)

Alpha 是以弧度表示的旋转角度,但我不知道是什么给出的img = img[:, ::-1])))


推荐阅读