ArrayFire: computer_vision/susan.cpp (original) (raw)

#include

#include

using namespace af;

static void susan_demo(bool console) {

if (console)

img_color = loadImage(ASSETS_DIR "/examples/images/square.png", true);

else

img_color = loadImage(ASSETS_DIR "/examples/images/man.jpg", true);

img_color /= 255.f;

features feat = susan(img, 3, 32.0f, 10, 0.05f, 3);

printf("No features found, exiting\n");

return;

}

float* h_x = feat.getX().host<float>();

float* h_y = feat.getY().host<float>();

const int draw_len = 3;

int x = h_x[f];

int y = h_y[f];

img_color(x, seq(y - draw_len, y + draw_len), 0) = 0.f;

img_color(x, seq(y - draw_len, y + draw_len), 1) = 1.f;

img_color(x, seq(y - draw_len, y + draw_len), 2) = 0.f;

img_color(seq(x - draw_len, x + draw_len), y, 0) = 0.f;

img_color(seq(x - draw_len, x + draw_len), y, 1) = 1.f;

img_color(seq(x - draw_len, x + draw_len), y, 2) = 0.f;

}

freeHost(h_x);

freeHost(h_y);

if (!console) {

while (!wnd.close()) wnd.image(img_color);

} else {

}

}

int main(int argc, char** argv) {

int device = argc > 1 ? atoi(argv[1]) : 0;

bool console = argc > 2 ? argv[2][0] == '-' : false;

try {

printf("** ArrayFire SUSAN Feature Detector Demo **\n\n");

susan_demo(console);

fprintf(stderr, "%s\n", ae.what());

throw;

}

return 0;

}

Window object to render af::arrays.

A multi dimensional data container.

T * host() const

Copy array data to host and return host pointer.

An ArrayFire exception class.

virtual const char * what() const

Returns an error message for the exception in a string format.

Represents a feature returned by a feature detector.

array getScore() const

Returns an array with the score of the features.

array getY() const

Returns an af::array which represents the y locations of a feature.

size_t getNumFeatures() const

Returns the number of features represented by this object.

array getX() const

Returns an af::array which represents the x locations of a feature.

seq is used to create sequences for indexing af::array

AFAPI void setDevice(const int device)

Sets the current device.

void image(const array &in, const char *title=NULL)

Renders the input array as an image to the window.

bool close()

Check if window is marked for close.