Function main

examples/image_processing/binary_thresholding.cpp:62–102

}

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

        array bimodal =
            loadImage(ASSETS_DIR "/examples/images/noisy_square.png", false);
        bimodal = resize(0.75f, bimodal);

        array bt         = threshold(bimodal, 180.0f);
        array ot         = otsu(bimodal);
        array bimodHist  = histogram(bimodal, 256, 0, 255);
        array smooth     = convolve(bimodal, gaussianKernel(5, 5));
        array smoothHist = histogram(smooth, 256, 0, 255);

        af::Window wnd(1536, 1024, "Binary Thresholding Algorithms");
        printf("Press ESC while the window is in focus to proceed to exit\n");

        wnd.grid(3, 3);
        wnd(0, 1).setAxesTitles("Bins", "Frequency");
        wnd(1, 1).setAxesTitles("Bins", "Frequency");
        wnd(2, 1).setAxesTitles("Bins", "Frequency");
        while (!wnd.close()) {
            wnd(0, 0).image(bimodal / 255, "Input Image");
            wnd(1, 0).image(bimodal / 255, "Input Image");
            wnd(2, 0).image(smooth / 255, "Input Smoothed by Gaussian Filter");
            wnd(0, 1).hist(bimodHist, 0, 255, "Input Histogram");
            wnd(1, 1).hist(bimodHist, 0, 255, "Input Histogram");
            wnd(2, 1).hist(smoothHist, 0, 255, "Smoothed Input Histogram");
            wnd(0, 2).image(bt, "Simple Binary threshold");
            wnd(1, 2).image(ot, "Otsu's Threshold");
            wnd(2, 2).image(otsu(smooth), "Otsu's Threshold on Smoothed Image");
            wnd.show();
        }
    } catch (af::exception& e) {
        fprintf(stderr, "%s\n", e.what());
        throw;
    }
    return 0;
}