#include <cstdio>
#include <cstdlib>
static void harris_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;
array idet = ixx * iyy - ixy * ixy;
array response = idet - 0.04f * (itr * itr);
array corners = response > 1e5f;
corners = corners * response;
corners = (corners == max_resp) * corners;
float* h_corners = corners.
host<
float>();
unsigned good_corners = 0;
const int draw_len = 3;
for (
int y = draw_len; y < img_color.
dims(0) - draw_len; y++) {
for (
int x = draw_len; x < img_color.
dims(1) - draw_len; x++) {
if (h_corners[x * corners.
dims(0) + y] > 1e5f) {
img_color(y,
seq(x-draw_len, x+draw_len), 0) = 0.f;
img_color(y,
seq(x-draw_len, x+draw_len), 1) = 1.f;
img_color(y,
seq(x-draw_len, x+draw_len), 2) = 0.f;
img_color(
seq(y-draw_len, y+draw_len), x, 0) = 0.f;
img_color(
seq(y-draw_len, y+draw_len), x, 1) = 1.f;
img_color(
seq(y-draw_len, y+draw_len), x, 2) = 0.f;
good_corners++;
}
}
}
printf("Corners found: %u\n", good_corners);
if (!console) {
} else {
const int good_corners = corners_x.
dims()[0];
std::cout << "Corners found: " << good_corners << std::endl << std::endl;
}
}
int main(int argc, char** argv)
{
int device = argc > 1 ? atoi(argv[1]) : 0;
bool console = argc > 2 ? argv[2][0] == '-' : false;
try {
std::cout << "** ArrayFire Harris Corner Detector Demo **" << std::endl << std::endl;
harris_demo(console);
std::cerr << ae.
what() << std::endl;
throw;
}
return 0;
}
Window object to render af::arrays.
Definition: graphics.h:37
A multi dimensional data container.
Definition: array.h:27
Definition: exception.h:20
virtual const char * what() const
Definition: exception.h:34
seq is used to create seq for indexing af::array
Definition: seq.h:46
@ AF_GRAY
Grayscale.
Definition: defines.h:296
@ AF_RGB
3-channel RGB
Definition: defines.h:297
AFAPI void grad(array &dx, array &dy, const array &in)
C++ Interface for calculating the gradients.
array constant(T val, const dim4 &dims, const dtype ty=(af_dtype) dtype_traits< T >::ctype)
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.
AFAPI array colorSpace(const array &image, const CSpace to, const CSpace from)
C++ Interface wrapper for colorspace conversion.
AFAPI array dilate(const array &in, const array &mask)
C++ Interface for image dilation (max filter)
AFAPI array gaussianKernel(const int rows, const int cols, const double sig_r=0, const double sig_c=0)
C++ Interface for generating gausian kernels.
AFAPI array loadImage(const char *filename, const bool is_color=false)
C++ Interface for loading an image.
dim4 dims() const
Get dimensions of the array.
T * host() const
Copy array data to host and return host pointer.
AFAPI array where(const array &in)
C++ Interface for finding the locations of non-zero values in an array.
AFAPI array convolve(const array &signal, const array &filter, const convMode mode=AF_CONV_DEFAULT, const convDomain domain=AF_CONV_AUTO)
C++ Interface for convolution any(one through three) dimensional signals.
Definition: algorithm.h:15
#define af_print(...)
Definition: util.h:141