在计算完单应性矩阵后,通过对两个图像进行偏移从而实现两个图像的全景拼接,如何保证图片不会因为过长而被切割,请给代码

在全景拼接过程中，为了防止图像被切割，需要对图像进行裁剪和缩放。具体的实现方法如下：

1. 计算图像的边界

可以通过单应性矩阵计算出两个图像的边界，然后取两个图像边界的交集作为拼接后图像的边界。

import cv2
import numpy as np

def get_image_boundary(image, H):
    """计算图像的边界"""
    h, w = image.shape[:2]
    # 计算图像四个角点的坐标
    pts = np.array([[0, 0, 1], [0, h-1, 1], [w-1, h-1, 1], [w-1, 0, 1]])
    # 计算变换后的坐标
    pts_transformed = np.dot(pts, H.T)
    pts_transformed = pts_transformed / pts_transformed[:, 2:]
    # 计算边界
    x_min = max(0, np.min(pts_transformed[:, 0]))
    x_max = min(w-1, np.max(pts_transformed[:, 0]))
    y_min = max(0, np.min(pts_transformed[:, 1]))
    y_max = min(h-1, np.max(pts_transformed[:, 1]))
    return int(x_min), int(x_max), int(y_min), int(y_max)

2. 裁剪图像

根据计算出的边界，可以对图像进行裁剪，保留拼接后的完整图像。

def crop_image(image, x_min, x_max, y_min, y_max):
    """裁剪图像"""
    return image[y_min:y_max+1, x_min:x_max+1, :]

3. 缩放图像

为了保证图像的分辨率一致，需要对图像进行缩放。可以根据裁剪后的图像大小和设定的输出图像大小，计算出缩放比例。

def resize_image(image, size):
    """缩放图像"""
    h, w = image.shape[:2]
    if h > w:
        scale = size / h
        new_h, new_w = size, int(w * scale)
    else:
        scale = size / w
        new_h, new_w = int(h * scale), size
    return cv2.resize(image, (new_w, new_h))

4. 完整代码

def stitch_images(image1, image2, H, output_size):
    """拼接两个图像"""
    # 计算图像1的边界
    x_min1, x_max1, y_min1, y_max1 = get_image_boundary(image1, np.identity(3, dtype=np.float))
    # 计算变换后图像2的边界
    x_min2, x_max2, y_min2, y_max2 = get_image_boundary(image2, H)
    # 计算两个图像的交集
    x_min = min(x_min1, x_min2)
    x_max = max(x_max1, x_max2)
    y_min = min(y_min1, y_min2)
    y_max = max(y_max1, y_max2)
    # 裁剪图像
    image1_cropped = crop_image(image1, x_min-x_min1, x_max-x_min1, y_min-y_min1, y_max-y_min1)
    image2_cropped = crop_image(image2, x_min-x_min2, x_max-x_min2, y_min-y_min2, y_max-y_min2)
    # 缩放图像
    image1_resized = resize_image(image1_cropped, output_size)
    image2_resized = resize_image(image2_cropped, output_size)
    # 拼接图像
    result = cv2.warpPerspective(image2_resized, H, output_size)
    result[y_min-y_min2:y_max-y_min2+1, x_min-x_min2:x_max-x_min2+1] = image1_resized
    return result