大家好，又见面了，我是你们的朋友全栈君。
原理：
实现：
/** * @description: 计算阶乘 * @param n 自然数 * @return 阶乘 */
int factorial(int n)
{
int fac = 1;
if (n == 0) return fac;
for (int i = 1; i <= n; ++i) fac *= i;
return fac;
}
/** * @description: 获得Sobel平滑算子 * @param size 掩膜大小 * @return Sobel平滑算子 */
cv::Mat getSobelSmooth(int size)
{
int n = size - 1;
cv::Mat SobelSmoothoper = cv::Mat::zeros(size, 1, CV_32F);
for (int k = 0; k <= n; k++)
{
float *pt = SobelSmoothoper.ptr<float>(0);
pt[k] = factorial(n) / (factorial(k)*factorial(n - k));
}
return SobelSmoothoper;
}
/** * @description: 获得Sobel差分算子 * @param size 掩膜大小 * @return Sobel差分算子 */
cv::Mat getSobeldiff(int size)
{
cv::Mat Sobeldiffoper = cv::Mat::zeros(cv::Size(size, 1), CV_32F);
cv::Mat SobelSmooth = getSobelSmooth(size - 1);
for (int k = 0; k < size; k++)
{
if (k == 0)
Sobeldiffoper.at<float>(0, k) = 1;
else if (k == size - 1)
Sobeldiffoper.at<float>(0, k) = -1;
else
Sobeldiffoper.at<float>(0, k) = SobelSmooth.at<float>(0, k) - SobelSmooth.at<float>(0, k - 1);
}
return Sobeldiffoper;
}
/** * @description: 卷积实现 * @param src 输入图像 * @param dst 输出图像 * @param kernel 卷积核 */
void conv2D(cv::Mat& src, cv::Mat& dst, cv::Mat kernel)
{
cv::flip(kernel, kernel, -1);
cv::filter2D(src, dst, src.depth(), kernel);
}
/** * @description: 可分离卷积———先水平方向卷积，后垂直方向卷积 * @param src 输入图像 * @param dst 输出图像 * @param kernel_X x方向卷积 * @param kernel_Y y方向卷积 */
void sepConv2D_X_Y(cv::Mat& src, cv::Mat& dst, cv::Mat kernel_X, cv::Mat kernel_Y)
{
cv::Mat dst_kernel_X;
conv2D(src, dst_kernel_X, kernel_X);
conv2D(dst_kernel_X, dst, kernel_Y);
}
/** * @description: 可分离卷积———先垂直方向卷积，后水平方向卷积 * @param src 输入图像 * @param dst 输出图像 * @param kernel_Y y方向卷积 * @param kernel_X x方向卷积 */
void sepConv2D_Y_X(cv::Mat& src, cv::Mat& dst, cv::Mat kernel_Y, cv::Mat kernel_X)
{
cv::Mat dst_kernel_Y;
conv2D(src, dst_kernel_Y, kernel_Y);
conv2D(dst_kernel_Y, dst, kernel_X);
}
/** * @description: Sobel算子边缘检测 * @param src 输入图像 * @param dst 输出图像 * @param size 掩膜大小 */
void sobel(cv::Mat& src, cv::Mat& dst, int size)
{
cv::Mat SobelSmoothoper = getSobelSmooth(size);
cv::Mat Sobeldiffoper = getSobeldiff(size);
cv::Mat dst_X, dst_Y;
sepConv2D_X_Y(src, dst_Y, SobelSmoothoper, Sobeldiffoper.t());
sepConv2D_Y_X(src, dst_X, SobelSmoothoper.t(), Sobeldiffoper);
dst = abs(dst_X) + abs(dst_Y);
cv::convertScaleAbs(dst, dst);
cv::convertScaleAbs(dst_X, dst_X);
cv::convertScaleAbs(dst_Y, dst_Y);
}
代码传送门：https://github.com/taifyang/OpenCV-algorithm
版权声明：本文内容由互联网用户自发贡献，该文观点仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容， 请联系我们举报，一经查实，本站将立刻删除。
发布者：全栈程序员-站长，转载请注明出处：https://javaforall.net/158976.html原文链接：https://javaforall.net
