Python实现索伯尔算子[通俗易懂]

全栈程序员-站长 • 2022年7月14日下午5:36 • 未分类 • 阅读 14

大家好，又见面了，我是你们的朋友全栈君。

Python实现索伯尔算子

最近在学习Python，正好用sobel算子练练手，将就看看吧
先放原图
在这里插入图片描述
用Opencv中sobel算子做一下对比：

# OpenCV
    ori_img = cv.imread("C:\\Users\\BLYX\\Desktop\\test\\temple1.jpg")
    x = cv.Sobel(ori_img[:, :, 0], cv.CV_16S, 1, 0)
    y = cv.Sobel(ori_img[:, :, 0], cv.CV_16S, 0, 1)
    absX = cv.convertScaleAbs(x)
    absY = cv.convertScaleAbs(y)
    dst = cv.addWeighted(absX, 0.5, absY, 0.5, 0)
    cv.imshow("OpenCV's Result", dst)

    cv.waitKey(0)
    cv.destroyAllWindows()

结果图如下

在这里插入图片描述

然后用自己写的sobel实现如下：

 # self—design

def convertu8(num):
    if num > 255 or num < -255:
        return 255
    elif -255 <= num <= 255:
        if abs(num - int(num)) < 0.5:
            return np.uint8(abs(num))
        else:
            return np.uint8(abs(num)) + 1


def sobel(img, k=0):
    row = img.shape[0]
    col = img.shape[1]
    image = np.zeros((row, col), np.uint8)
    s = time.time()
    for i in range(1, row - 1):
        for j in range(1, col - 1):
            y = int(img[i - 1, j + 1, k]) - int(img[i - 1, j - 1, k]) + 2 * (
                        int(img[i, j + 1, k]) - int(img[i, j - 1, k])) + int(img[i + 1, j + 1, k]) - int(
                img[i + 1, j - 1, k])
            x = int(img[i + 1, j - 1, k]) - int(img[i - 1, j - 1, k]) + 2 * (
                        int(img[i + 1, j, k]) - int(img[i - 1, j, k])) + int(img[i + 1, j + 1, k]) - int(
                img[i - 1, j + 1, k])
            image[i, j] = convertu8(abs(x) * 0.5 + abs(y) * 0.5)
    e = time.time()
    print(e - s)
    return image


if __name__ == '__main__':
    ori_img = cv.imread("C:\\Users\\BLYX\\Desktop\\test\\temple1.jpg")
    sobelimage = sobel(ori_img, 0)
    cv.imshow("my Result", sobelimage)
    cv.waitKey(0)
    cv.destroyAllWindows()

结果图如下
在这里插入图片描述
然后用ENVI比对了一下：

在这里插入图片描述
***blablablabla
可以看到三幅图相对比，第二幅图的视觉效果更好，当然不同的图片可能会对最终的结果产生影响。但经多组图片测试，可以得到在对水平锐化和垂直锐化图像赋予相同的权重叠合时，第二幅结果图在视觉上稍微优于OpenCV中的sobel，而在处理效率上却要低上很多。希望未来能够改进一下。
写得不好，还望大家指正！