1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
|
#include <cstdlib>
#include <cstdio>
#include <opencv/cv.hpp>
#include <opencv2/highgui/highgui.hpp>
#include <opencv2/contrib/contrib.hpp>
#define err(fmt, args...) fprintf(stderr, "Error: " fmt, ##args)
int main(int argc, char **argv)
{
int ret = 0;
CvCapture *cap = cvCaptureFromCAM(CV_CAP_ANY);
if (!cap) {
err("Failed to open camera\n");
exit(EXIT_FAILURE);
}
/* get initial frame to determine width/height */
IplImage *frame = cvQueryFrame(cap);
if (!frame) {
err("Failed to capture frame\n");
exit(EXIT_FAILURE);
}
int w = frame->width;
int h = frame->height;
printf("Image size is %dx%d\n", w, h);
int ro0 = 3; /* radius of blind spot */
int R = 240; /* no. of rings */
cv::Point2i c(w / 2.0, h / 2.0);
cv::LogPolar_Adjacent lp_adj(w, h, c, R, ro0, 0.25);
cv::LogPolar_Overlapping lp_overlap(w, h, c, R, ro0);
cv::LogPolar_Interp lp_bilin(w, h, c, R, ro0);
cv::LogPolar_Interp lp_nearest(w, h, c, R, ro0, cv::INTER_NEAREST);
cvNamedWindow("retinal", CV_WINDOW_AUTOSIZE);
cv::Mat cortical, retinal;
int key = 'a', i = 0;
printf("Available modes: [a]djacent, [o]verlap, [b]ilniear, [n]earest\n");
printf("Starting to capture, press 'q' to exit...\n");
time_t start, end;
int steps = 0;
time(&start);
while (1) {
frame = cvQueryFrame(cap);
if (!frame) {
err("Failed to capture frame\n");
ret = -1;
break;
}
cv::Mat img(frame);
if (key == 'a') {
cv::cvtColor(img, img, CV_BGR2GRAY);
cortical = lp_adj.to_cortical(img);
retinal = lp_adj.to_cartesian(cortical);
} else if (key == 'o') {
cortical = lp_overlap.to_cortical(img);
retinal = lp_overlap.to_cartesian(cortical);
} else if (key == 'b') {
cortical = lp_bilin.to_cortical(img);
retinal = lp_bilin.to_cartesian(cortical);
} else if (key == 'n') {
cortical = lp_nearest.to_cortical(img);
retinal = lp_nearest.to_cartesian(cortical);
}
time(&end);
steps++;
double s = difftime(end, start);
double fps = steps / s;
char buf[32];
snprintf(buf, sizeof(buf), "FPS: %lf", fps);
putText(img, buf, cv::Point(30, 30),
cv::FONT_HERSHEY_COMPLEX_SMALL, 0.8,
cv::Scalar(200,200,250), 1, CV_AA);
imshow("retinal", retinal);
int c = cvWaitKey(10);
if ((c & 255) == 'p') {
char name[128];
snprintf(name, sizeof(name), "retinal-%d.jpg", i);
imwrite(name, retinal);
printf("Retinal image written to %s...\n", name);
snprintf(name, sizeof(name), "cortical-%d.jpg", i);
imwrite(name, cortical);
printf("Cortical image written to %s...\n", name);
snprintf(name, sizeof(name), "diff-%d.jpg", i);
imwrite(name, img - cortical);
printf("Difference image written to %s...\n", name);
i++;
} else if (c > 0)
key = c;
if (key == 'q' || (key & 255) == 27)
break;
}
return ret;
}
|
