Imgaug获取透视变换后的坐标
时间: 2023-09-15 19:19:59 浏览: 140
使用imgaug进行透视变换后,可以通过以下代码获取变换后的坐标:
```python
import numpy as np
import imgaug.augmenters as iaa
# 定义透视变换器
perspective_transform = iaa.PerspectiveTransform(scale=(0.01, 0.15))
# 生成一个随机图片
image = np.random.randint(0, 255, (100, 100, 3), dtype=np.uint8)
# 对图片进行透视变换
image_aug, matrix = perspective_transform.augment(image=image, return_matrix=True)
# 获取变换后的四个角点坐标
h, w = image.shape[:2]
corners = np.float32([[0, 0], [0, h], [w, h], [w, 0]])
corners_aug = cv2.perspectiveTransform(corners.reshape(-1, 1, 2), matrix).reshape(-1, 2)
print(corners_aug)
```
这里使用`PerspectiveTransform`定义了一个透视变换器,然后使用`augment`方法对图片进行变换,并返回变换矩阵`matrix`。最后使用`perspectiveTransform`函数将原图像的四个角点坐标映射到变换后的图像中,从而获得变换后的四个角点坐标`corners_aug`。
阅读全文