如何消除griddata插值层析成像后扭曲图像边缘的彩虹伪影

伪影产生的核心原因 #

• 三次（cubic）插值存在天然的数值过冲特性，插值结果可能超出0~255的像素合法值域，直接将结果转为uint8类型时，溢出的数值会发生截断错位，直接导致色值错乱形成彩虹伪影
• 原代码对变换后越界的坐标直接做了clip操作，大量原本不属于目标图像边缘的无效像素被硬塞到边缘位置，插值时这些异常值会被纳入计算，进一步放大边缘色偏
• 前向映射本身存在点重叠的问题，同一目标坐标可能被多个源像素重复赋值，也会引入异常色值

代码调整方案 #

针对上述问题，修改现有warp方法的对应逻辑即可，核心改动三点：

1. 移除强制clip坐标的逻辑，先筛选所有落在目标图像合法范围内的坐标，直接丢弃越界点，避免无效值污染边缘
2. 插值结束后先把数值截断到0~255区间，再转为uint8类型，彻底解决数值溢出问题
3. 可选：如果边缘对平滑度要求不高，可以把三次插值换成线性插值，过冲效应会大幅降低，边缘效果更稳定

修改后的完整代码如下：

@staticmethod
def warp(
        backward_projective_homography: np.ndarray,
        src_image: np.ndarray,
        dst_image_shape: tuple = (1088, 1452, 3)) -> np.ndarray:

    x_src, y_src = np.arange(0, src_image.shape[1]), np.arange(0, src_image.shape[0])
    coord_mat_src = np.dstack(np.meshgrid(x_src, y_src)).reshape(-1, 2).transpose()
    ones_src = np.ones(coord_mat_src.shape[1]).astype(int)
    coord_mat_ones_src = np.vstack([coord_mat_src, ones_src])

    back_homo_dst = np.dot(backward_projective_homography, coord_mat_ones_src)
    z_row = back_homo_dst[-1,:]
    dst_coord_normalized = back_homo_dst / np.array([z_row,]*3)
    
    # 筛选合法坐标，丢弃越界点，不再硬clip到边缘
    x_dest_coord = dst_coord_normalized[0,:]
    y_dest_coord = dst_coord_normalized[1,:]
    valid_mask = (x_dest_coord >= 0) & (x_dest_coord < dst_image_shape[1]) & \
                 (y_dest_coord >= 0) & (y_dest_coord < dst_image_shape[0])
    # 仅保留合法范围内的点参与后续计算
    x_dest_valid = x_dest_coord[valid_mask].astype(int)
    y_dest_valid = y_dest_coord[valid_mask].astype(int)
    src_valid_coord = coord_mat_src[:, valid_mask].astype(int)

    points = np.vstack([y_dest_valid, x_dest_valid]).transpose()
    values = src_image[src_valid_coord[1, :], src_valid_coord[0, :], :].astype(np.float32)

    grid_x, grid_y = np.mgrid[0:dst_image_shape[0], 0:dst_image_shape[1]]
    grid_image = griddata(points, values, (grid_x, grid_y), method='cubic')
    
    # 先截断数值到合法像素范围再转uint8，避免溢出导致的色值错乱
    grid_image = np.clip(grid_image, 0, 255).astype(np.uint8)
    return grid_image

如果调整后边缘仍有轻微伪影，将method='cubic'改为method='linear'即可。

内容的提问来源于stack exchange，提问作者Guy