C++ OpenCV 实战：频率域图像增强及代码示例与注意要点

好的，下面是一个使用 C++ 和 OpenCV 实现频率域图像增强的示例代码。这个例子展示了如何使用傅里叶变换对图像进行频域滤波，具体来说是高通滤波器。 ```cpp #include <opencv2/opencv.hpp> #include <iostream> // 高通滤波器函数 void highPassFilter(cv::Mat &input, cv::Mat &output, int cutoff) { // 计算中心位置 int cx = input.cols / 2; int cy = input.rows / 2; // 创建与输入相同大小的掩码 cv::Mat mask = cv::Mat::zeros(input.size(), CV_32F); // 设置高通滤波器 for (int i = 0; i < input.rows; ++i) { for (int j = 0; j < input.cols; ++j) { double distance = std::sqrt(std::pow(i - cy, 2) + std::pow(j - cx, 2)); if (distance > cutoff) { mask.at<float>(i, j) = 1.0; } } } // 应用掩码 cv::multiply(input, mask, output); } int main() { // 读取图像 cv::Mat image = cv::imread("input.jpg", cv::IMREAD_GRAYSCALE); if (image.empty()) { std::cerr << "无法读取图像" << std::endl; return -1; } // 将图像转换为浮点类型 cv::Mat floatImage; image.convertTo(floatImage, CV_32F); // 进行傅里叶变换 cv::Mat complexImage; cv::dft(floatImage, complexImage, cv::DFT_COMPLEX_OUTPUT); // 将频谱移到图像中心 cv::Mat shiftedSpectrum; cv::dft(complexImage, shiftedSpectrum, cv::DFT_SCALE | cv::DFT_CENTER_ZERO); // 应用高通滤波器 cv::Mat filteredSpectrum; highPassFilter(shiftedSpectrum, filteredSpectrum, 30); // 将频谱移回原位 cv::Mat unshiftedSpectrum; cv::dft(filteredSpectrum, unshiftedSpectrum, cv::DFT_INVERSE | cv::DFT_REAL_OUTPUT); // 将结果转换回8位图像 cv::Mat result; unshiftedSpectrum.convertTo(result, CV_8U); // 显示结果 cv::imshow("原始图像", image); cv::imshow("增强后的图像", result); cv::waitKey(0); // 保存结果 cv::imwrite("enhanced_image.jpg", result); return 0; } ``` 额外需要注意的逻辑点 1. 边界处理：在创建掩码时，确保边界条件处理得当，避免越界访问。 2. 参数选择：`cutoff` 参数的选择会影响滤波效果，需要根据实际需求进行调整。 3. 性能优化：对于大图像，可以考虑使用多线程或 GPU 加速来提高处理速度。 希望这段代码对你有帮助！如果有任何问题，欢迎随时提问。 ######[AI写代码神器 | 724点数解答 | 2024-12-07 12:24:59]