canny() [2/2]

MAYAFLUX_API std::vector< float > MayaFlux::Kinesis::Vision::canny	(	std::span< const float >	gray,
		uint32_t	w,
		uint32_t	h,
		float	sigma,
		float	low_threshold,
		float	high_threshold
	)

Canny edge detector.

Pipeline: gaussian_blur(sigma) -> sobel -> non-maximum suppression -> double threshold -> hysteresis. Output is binary: 1.0f on edges, 0.0f elsewhere.

Parameters

gray	Single-channel float span, size must be w * h.
w	Image width in pixels.
h	Image height in pixels.
sigma	Gaussian pre-blur sigma. 0.0f skips blur.
low_threshold	Hysteresis low threshold in [0, 1].
high_threshold	Hysteresis high threshold in [0, 1].

Returns

Binary float vector of length w * h.

Definition at line 130 of file Gradient.cpp.

{
    const size_t n = static_cast<size_t>(w) * h;
 
    std::vector<float> blurred;
    std::span<const float> src = gray;
    if (sigma > 0.0F) {
        blurred = gaussian_blur(gray, w, h, sigma);
        src = blurred;
    }
 
    auto grad = sobel(src, w, h);
 
    std::vector<float> nms(n, 0.0F);
    std::for_each(std::execution::par_unseq,
        std::views::iota(size_t { 0 }, n).begin(),
        std::views::iota(size_t { 0 }, n).end(),
        [&](size_t idx) {
            const auto px = static_cast<int32_t>(idx % w);
            const auto py = static_cast<int32_t>(idx / w);
 
            if (px == 0 || py == 0
                || px == static_cast<int32_t>(w) - 1
                || py == static_cast<int32_t>(h) - 1)
                return;
 
            const float angle = grad.angle[idx];
            const float pi = std::numbers::pi_v<float>;
 
            float norm_angle = angle < 0.0F ? angle + pi : angle;
            norm_angle = norm_angle / pi * 4.0F;
 
            int32_t ox = 0, oy = 0;
            if (norm_angle < 0.5F || norm_angle >= 3.5F) {
                ox = 1;
            } else if (norm_angle < 1.5F) {
                ox = 1;
                oy = 1;
            } else if (norm_angle < 2.5F) {
                oy = 1;
            } else {
                ox = -1;
                oy = 1;
            }
 
            const float m = grad.magnitude[idx];
            const float m_pos = grad.magnitude[static_cast<size_t>(py + oy) * w + static_cast<size_t>(px + ox)];
            const float m_neg = grad.magnitude[static_cast<size_t>(py - oy) * w + static_cast<size_t>(px - ox)];
 
            if (m >= m_pos && m >= m_neg)
                nms[idx] = m;
        });
 
    std::vector<float> out(n, 0.0F);
    std::vector<uint8_t> strong(n, 0);
    std::vector<uint8_t> weak(n, 0);
 
    for (size_t i = 0; i < n; ++i) {
        if (nms[i] >= high_threshold) {
            strong[i] = 1;
        } else if (nms[i] >= low_threshold) {
            weak[i] = 1;
        }
    }
 
    std::vector<size_t> stack;
    for (size_t i = 0; i < n; ++i) {
        if (strong[i])
            stack.push_back(i);
    }
 
    while (!stack.empty()) {
        const size_t idx = stack.back();
        stack.pop_back();
        out[idx] = 1.0F;
 
        const auto px = static_cast<int32_t>(idx % w);
        const auto py = static_cast<int32_t>(idx / w);
 
        for (int32_t dy = -1; dy <= 1; ++dy) {
            for (int32_t dx = -1; dx <= 1; ++dx) {
                if (dx == 0 && dy == 0)
                    continue;
                const int32_t nx = px + dx;
                const int32_t ny = py + dy;
                if (nx < 0 || ny < 0
                    || nx >= static_cast<int32_t>(w)
                    || ny >= static_cast<int32_t>(h))
                    continue;
                const size_t ni = static_cast<size_t>(ny) * w + nx;
                if (weak[ni] && out[ni] == 0.0F)
                    stack.push_back(ni);
            }
        }
    }
 
    return out;
}

References gaussian_blur(), h, out, px, py, sobel(), src, and stack.

Referenced by canny(), and MayaFlux::Kinesis::Vision::VisionExecutor::run().

Here is the call graph for this function:

Here is the caller graph for this function: