Function main

examples/image_processing/gradient_diffusion.cpp:51–98

}

int main(int argc, char *argv[]) {
    int device = argc > 1 ? atoi(argv[1]) : 0;

    try {
        setDevice(device);
        info();

        printf("** ArrayFire Gradient Anisotropic Smoothing Demo **\n");

        Window myWindow("Gradient Anisotropic Smoothing");

        in = loadImage(ASSETS_DIR "/examples/images/man.jpg", false);

        array sEdges = sobelFilter(in);

        anisotropicSmoothing();

        array Gx, Gy;
        sobel(Gx, Gy, smoothed, 3);

        edges = normalize(hypot(Gx, Gy));

        while (!myWindow.close()) {
            myWindow.grid(2, 2);

            myWindow(0, 0).image(in / 255.0f, "Input Image");
            myWindow(0, 1).image(normalize(smoothed),
                                 "Anisotropically smooted Input");
            myWindow(1, 0).image(normalize(sEdges),
                                 "Gradient Magnitude after gaussian blur t=64");
            myWindow(1, 1).image(normalize(edges),
                                 "Gradient Magnitude after diffusion t=64");

            myWindow.show();
        }

        printf(
            "\nAnisotropic Diffusion avg runtime for current image in Seconds: "
            "%g\n",
            timeit(anisotropicSmoothing));

    } catch (af::exception &e) {
        fprintf(stderr, "%s\n", e.what());
        throw;
    }

    return 0;
}