Function edge

examples/image_processing/edge.cpp:46–69

}

array edge(const array &in, int method = 0) {
    int w = 5;
    if (in.dims(0) < 512) w = 3;
    if (in.dims(0) > 2048) w = 7;

    int h = 5;
    if (in.dims(0) < 512) h = 3;
    if (in.dims(0) > 2048) h = 7;

    array ker    = gaussianKernel(w, h);
    array smooth = convolve(in, ker);
    array mag, dir;

    switch (method) {
        case 1: prewitt(mag, dir, smooth); break;
        case 2: sobelFilter(mag, dir, smooth); break;
        case 3:
            mag = canny(in, AF_CANNY_THRESHOLD_AUTO_OTSU, 0.18, 0.54).as(f32);
            break;
        default: throw af::exception("Unsupported type");
    }

    return normalize(mag);
}

void edge() {
    af::Window myWindow("Edge Dectectors");