fft_upsampling

函数功能

对图像进行相位调整和插值，以实现上采样，即放大图像。

将原始图片矩阵经过FFT转变成频域上的信息矩阵。
将频域矩阵按照shift进行相位的调整。
对频域矩阵进行周期性的填充（高频补0），补充至目标矩阵大小。
经过iFFT将频域矩阵转换回空域矩阵，完成图像插值。

函数接口

void kcv_FFTUpsampling_acc(const double *_src, double *_dst, size_t src_rows, size_t src_cols, size_t dst_rows, size_t dst_cols, double shift_x, double shift_y);

参数

参数名	描述	取值范围	输入/输出
src	输入图像。	非空且申请内存大小为src_rowssrc_colssizeof(double)	输入
dst	插值后的输出图像。	非空且申请内存大小为dst_rowsdst_colssizeof(double)	输出
src_rows	输入图像行数。	(0, 16384]	输入
src_cols	输入图像列数。	(0, 16384]	输入
dst_rows	输出图像行数。	[src_rows, 16384]	输出
dst_cols	输出图像列数。	[src_cols, 16384]	输出
shift_x	x方向偏移量，输出图像原点相比于输入图像。	(-inf, inf)	输出
shift_y	y方向偏移量，输出图像原点相比于输入图像。	(-inf, inf)	输出

返回值

成功：返回LITHO_IMG_STS_NO_ERR。
失败：返回错误码。

错误码

错误码	描述
LITHO_IMG_STS_NULL_PTR_ERR	参数“src”或“dst”存在空指针。
LITHO_IMG_STS_SIZE_ERR	参数“src_rows”或“src_cols”值等于0。参数“dst_rows”值小于“src_rows”，或“dst_cols”值小于“src_cols”。
LITHO_IMG_STS_FFT_ERR	FFT plan创建失败。

示例

#include <stdio.h>
#include <stdlib.h>
#include <complex.h>
#include "kp_litho_img.h"

int main() {
    // 定义输入图像的尺寸和目标尺寸
    size_t src_rows = 5;
    size_t src_cols = 5;
    size_t dst_rows = 10;
    size_t dst_cols = 10;
    // 定义输入图像和输出图像的指针
    double *src = (double *)malloc(src_rows * src_cols * sizeof(double));
    double *dst = (double *)malloc(dst_rows * dst_cols * sizeof(double));
    // 初始化输入图像（这里使用一个简单的5x5图像）
    double src_data[5][5] = {
        {0, 0, 0, 0, 0},
        {0, 1, 1, 1, 0},
        {0, 1, 0, 1, 0},
        {0, 1, 1, 1, 0},
        {0, 0, 0, 0, 0}
    };
    for (size_t i = 0; i < src_rows; i++) {
        for (size_t j = 0; j < src_cols; j++) {
            src[i * src_cols + j] = src_data[i][j];
        }
    }
    // 定义相位调整参数
    double shift_x = 0.5;
    double shift_y = 0.5;
    // 调用kcv_FFTUpsampling_acc函数进行上采样
    cv::kcv_FFTUpsampling_acc(src, dst, src_rows, src_cols, dst_rows, dst_cols, shift_x, shift_y);
    // 打印结果（这里只打印部分结果）
    printf("Upsampled Image:\n");
    for (size_t i = 0; i < dst_rows; i++) {
        for (size_t j = 0; j < dst_cols; j++) {
            printf("%.2f ", dst[i * dst_cols + j]);
            if (j == dst_cols - 1) {
                printf("\n");
            }
        }
    }
    // 释放内存
    free(src);
    free(dst);
    return 0;
}

运行结果：

Upsampled Image:
02 0.05 0.09 0.11 0.11 0.09 0.05 0.02 -0.01 -0.01 
05 0.17 0.24 0.26 0.26 0.24 0.17 0.05 -0.04 -0.04 
09 0.24 0.25 0.20 0.20 0.25 0.24 0.09 -0.07 -0.07 
11 0.26 0.20 0.05 0.05 0.20 0.26 0.11 -0.09 -0.09 
11 0.26 0.20 0.05 0.05 0.20 0.26 0.11 -0.09 -0.09 
09 0.24 0.25 0.20 0.20 0.25 0.24 0.09 -0.07 -0.07 
05 0.17 0.24 0.26 0.26 0.24 0.17 0.05 -0.04 -0.04 
02 0.05 0.09 0.11 0.11 0.09 0.05 0.02 -0.01 -0.01 
-0.01 -0.04 -0.07 -0.09 -0.09 -0.07 -0.04 -0.01 0.01 0.01 
-0.01 -0.04 -0.07 -0.09 -0.09 -0.07 -0.04 -0.01 0.01 0.01

父主题： 接口定义