25 #include <classifiers/surf.h> 26 #include <filters/roidraw.h> 27 #include <fvutils/color/colorspaces.h> 28 #include <fvutils/readers/jpeg.h> 29 #include <fvutils/readers/png.h> 30 #include <fvutils/writers/png.h> 31 #include <opencv/cv.h> 32 #include <opencv/cxcore.h> 33 #include <opencv/highgui.h> 38 main(
int argc,
char **argv)
41 printf(
"Usage: %s <object-image-file.png> <scene-image-file.png>\n", argv[0]);
45 const char *object_file = argv[1];
46 const char *scene_file = argv[2];
48 printf(
"QASurfClassifier: creating cvImages for object and scene\n");
49 IplImage *obj_img = cvLoadImage(object_file, 1);
50 IplImage *scn_img = cvLoadImage(scene_file, 1);
53 printf(
"QASurfClassifier: Load scene as image\n");
57 PNGReader * reader =
new PNGReader(scene_file);
58 unsigned char *buffer =
59 malloc_buffer(YUV422_PLANAR, reader->pixel_width(), reader->pixel_height());
60 reader->set_buffer(buffer);
67 printf(
"QASurfClassifier: Instantiate SurfClassifier\n");
68 SurfClassifier *classifier =
new SurfClassifier(object_file);
70 classifier->set_src_buffer(buffer, reader->pixel_width(), reader->pixel_height());
72 printf(
"QASurfClassifier: classify ...\n");
73 std::list<ROI> *rois = classifier->classify();
75 printf(
"QASurfClassifier: filterROI\n");
76 FilterROIDraw *roi_draw =
new FilterROIDraw();
77 for (std::list<ROI>::iterator i = rois->begin(); i != rois->end(); ++i) {
78 printf(
"QASurfClassifier: ROI: start (%u, %u) extent %u x %u\n",
84 roi_draw->set_dst_buffer(buffer, &(*i));
88 printf(
"QASurfClassifier: draw ROIs in cvWindow\n");
89 for (std::list<ROI>::iterator i = rois->begin(); i != rois->end(); ++i) {
90 if ((*i).height == 11 && (*i).width == 11) {
92 cvPoint((*i).start.x, (*i).start.y),
93 cvPoint((*i).start.x + (*i).width, (*i).start.y + (*i).height),
99 cvPoint((*i).start.x, (*i).start.y),
100 cvPoint((*i).start.x + (*i).width, (*i).start.y + (*i).height),
108 cvNamedWindow(
"Scene-Matches", 1);
109 cvShowImage(
"Scene-Matches", scn_img);
110 cvNamedWindow(
"Object", 1);
111 cvShowImage(
"Object", obj_img);