CUFFT库与奇数尺寸图像FFTShift操作异常问题求助

我正在为考试实现基于GPU的图像缩放代码，采用C++语言与CUFFT库执行快速傅里叶变换（FFT），目标是支持处理奇偶尺寸的图像。当前测试的基础流程（暂未加入缩放相关的掩码裁剪步骤）为：

• 使用CUFFT执行正向FFT
• 将频谱低频分量移至中心（FFTShift）
• 执行逆FFTShift（将频谱移回原始位置）
• 使用CUFFT执行逆IFFT

但在处理奇数尺寸的正方形图像时，仅执行上述流程后，输出图像的颜色与原图明显不符。暂时搁置图像缩放的初始需求，优先解决该基础FFT/IFFT流程的问题。

相关代码片段：

// 正向FFT初始化与执行
cufftHandle fftPlan;
cufftPlan2d(&fftPlan, imgHeight, imgWidth, CUFFT_C2C);
cufftExecC2C(fftPlan, d_complexInput, d_complexFFTOut, CUFFT_FORWARD);

// FFTShift核调用（自定义实现）
dim3 blockDim(16, 16);
dim3 gridDim((imgWidth + blockDim.x - 1)/blockDim.x, (imgHeight + blockDim.y - 1)/blockDim.y);
fftShiftKernel<<<gridDim, blockDim>>>(d_complexFFTOut, imgWidth, imgHeight);

// 逆FFTShift核调用
ifftShiftKernel<<<gridDim, blockDim>>>(d_complexFFTOut, imgWidth, imgHeight);

// 逆FFT执行
cufftExecC2C(fftPlan, d_complexFFTOut, d_complexIFFTOut, CUFFT_INVERSE);

// 设备到主机拷贝与图像格式转换
cudaMemcpy(h_complexResult, d_complexIFFTOut, imgWidth*imgHeight*sizeof(cufftComplex), cudaMemcpyDeviceToHost);
// ... 转换为RGB/灰度图像并输出

输入输出图像：

输入图像：奇数尺寸正方形原图
输出图像：颜色异常的处理后图像

• FFTShift/IFFTShift的奇数尺寸逻辑错误
  对于奇数尺寸N，FFTShift的偏移量应为(N//2, N//2)（整数除法），若实现中错误使用浮点偏移或索引计算未考虑奇数边界，会导致频谱移位不完整。需检查CUDA核函数中元素交换的索引逻辑，确保每个元素仅被交换一次，避免重复操作或遗漏。

• 逆FFT后未做正确归一化
  CUFFT的逆IFFT结果不会自动归一化，必须手动除以总像素数imgHeight * imgWidth。奇数尺寸下总像素数为奇数，需使用浮点除法，避免整数除法导致结果截断，进而引发颜色偏差。

• 多通道图像的处理逻辑错误
  若处理RGB图像，需对每个通道单独执行FFT/IFFT流程。如果错误地将三通道数据合并为单复数通道处理，会直接导致颜色混乱。需确认是否为每个通道分配了独立的复数数组，并分别执行变换。

• CUFFT计划与实际尺寸不匹配
  检查创建CUFFT计划时传入的imgHeight和imgWidth是否为图像实际的奇数尺寸，若计划尺寸与数据尺寸不一致，会导致变换结果完全错误。

• 复数数组初始化或内存拷贝错误
  确认复数数组的虚部初始化为0，未初始化的虚部会引入噪声；同时检查cudaMemcpy的字节数计算是否正确（imgWidth*imgHeight*sizeof(cufftComplex)），避免数据拷贝不完整。

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