A high-performance general-purpose compute library

- Main Page
- Tutorials
- Release Notes
- Examples

image_processing/confidence_connected_components.cpp
/*******************************************************
* Copyright (c) 2019, ArrayFire
* All rights reserved.
*
* This file is distributed under 3-clause BSD license.
* The complete license agreement can be obtained at:
* http://arrayfire.com/licenses/BSD-3-Clause
********************************************************/
#include <arrayfire.h>
#include <cassert>
#include <cstdio>
#include <cstdlib>
#include <iostream>
using namespace af;
array normalize01(const array& in) {
float min = af::min<float>(in);
float max = af::max<float>(in);
return (in - min) / (max - min);
}
void markCrossHair(array& in, const unsigned x, const unsigned y,
const float val) {
const int draw_len = 5;
for (int i = -1; i < 2; i++) {
in(x + i, seq(y - draw_len, y + draw_len), 0) = val;
in(x + i, seq(y - draw_len, y + draw_len), 1) = 0.f;
in(x + i, seq(y - draw_len, y + draw_len), 2) = 0.f;
in(seq(x - draw_len, x + draw_len), y + i, 0) = val;
in(seq(x - draw_len, x + draw_len), y + i, 1) = 0.f;
in(seq(x - draw_len, x + draw_len), y + i, 2) = 0.f;
}
}
int main(int argc, char* argv[]) {
try {
unsigned radius = 3;
unsigned multiplier = 2;
int iter = 3;
array input =
loadImage(ASSETS_DIR "/examples/images/depression.jpg", false);
array normIn = normalize01(input);
unsigned seedx = 162;
unsigned seedy = 126;
array blob = confidenceCC(input, 1, &seedx, &seedy, radius, multiplier,
iter, 255);
array colorIn = colorSpace(normIn, AF_RGB, AF_GRAY);
array colorOut = colorSpace(blob, AF_RGB, AF_GRAY);
markCrossHair(colorIn, seedx, seedy, 1);
markCrossHair(colorOut, seedx, seedy, 255);
af::Window wnd("Confidence Connected Components Demo");
while (!wnd.close()) {
wnd.grid(1, 2);
wnd(0, 0).image(colorIn, "Input Brain Scan");
wnd(0, 1).image(colorOut, "Region connected to Seed(162, 126)");
wnd.show();
}
} catch (af::exception& e) {
fprintf(stderr, "%s\n", e.what());
throw;
}
return 0;
}
arrayfire.h
af::Window
Window object to render af::arrays.
Definition graphics.h:37
af::array
A multi dimensional data container.
Definition array.h:37
af::exception
An ArrayFire exception class.
Definition exception.h:22
af::exception::what
virtual const char * what() const
Returns an error message for the exception in a string format.
Definition exception.h:46
af::seq
seq is used to create sequences for indexing af::array
Definition seq.h:46
AF_GRAY
@ AF_GRAY
Grayscale.
Definition defines.h:340
AF_RGB
@ AF_RGB
3-channel RGB
Definition defines.h:341
af::colorSpace
AFAPI array colorSpace(const array &image, const CSpace to, const CSpace from)
C++ Interface wrapper for colorspace conversion.
af::confidenceCC
AFAPI array confidenceCC(const array &in, const array &seeds, const unsigned radius, const unsigned multiplier, const int iter, const double segmentedValue)
C++ Interface for confidence connected components.
af::loadImage
AFAPI array loadImage(const char *filename, const bool is_color=false)
C++ Interface for loading an image.
af::max
AFAPI array max(const array &in, const int dim=-1)
C++ Interface to return the maximum along a given dimension.
af::min
AFAPI array min(const array &in, const int dim=-1)
C++ Interface to return the minimum along a given dimension.
af
Definition algorithm.h:15