Julia中的Sobel算子

Question

我是朱莉娅编程的初学者，我想实现Sobel运算符。不幸的是，以下代码的输出只是一个黑色的图像。

using Images, Colors, FileIO, Plots;
img = load("Lenna.png");
img_gray = Gray.(img);
sobel_image = convert(Array{Float64}, img_gray);
kernel_x =  Array{Float64}([1 0 -1; 2 0 -2; 1 0 -1]);
kernel_y =  Array{Float64}([1 2 1; 0 0 0; -1 -2 -1]);

#plot(img_gray)
function sobel(img)
    edge_img = zeros(Gray{Float64}, size(img, 1), size(img, 2));
    for x in 1:size(edge_img, 1) - size(kernel_x,1)
        for y in 1:size(edge_img, 2) - size(kernel_y,2)
            gx = sum(Gray{Float64}.(
                @view edge_img[x:x+size(kernel_x,1)-1, y:y+size(kernel_y,2)-1]) .* kernel_x)
            gy = sum(Gray{Float64}.(
                    @view edge_img[x:x+size(kernel_x,1)-1, y:y+size(kernel_y,2)-1]) .* kernel_y)
            edge_img[x+1, y+1] = hypot(gx, gy)
        end
    end
    return edge_img;
end

last_image = sobel(sobel_image)
plot(last_image)

我把kernel_x和kernel_y转换成Array64。基本思想是将代码从python转换为Julia，因为我不知道语法。你能帮我拿些小费吗？谢谢!

Answer 1

您的代码可以使用一些编辑(不必要的类型调用、全局变量等)，但我不认为它对错误结果负有太多责任。

它看起来像您的sobel函数接受一个img，但它只在第一行中使用它来创建一个黑色(全部为0)的edge_img，然后只对函数的其余部分使用edge_img。我想这就是为什么你最后会有一个黑色的形象。也许您打算在img和gy = ...行中索引来自gy = ...的值？

Answer 2

您可以尝试使用ImageFiltering

using ImageFiltering, TestImages, Plots
im = testimage("cameraman")
kernel = [1 0 -1; 2 0 -2;1 0 -1]
ims = imfilter(im, kernel)

Plots.plot(hcat(im, ims))

​

​