diff options
| -rw-r--r-- | Makefile | 15 | ||||
| -rw-r--r-- | logpolar.cpp | 114 |
2 files changed, 129 insertions, 0 deletions
diff --git a/Makefile b/Makefile new file mode 100644 index 0000000..d2c3d30 --- /dev/null +++ b/Makefile @@ -0,0 +1,15 @@ +P = logpolar + +CXXFLAGS += $(shell pkg-config --cflags opencv) +LDFLAGS += $(shell pkg-config --libs opencv) + +all: $(P) + +$(P): $(P).o + $(CXX) -o $@ $^ $(LDFLAGS) + +%.o: %.cpp + $(CXX) $(CXXFLAGS) -c $< -o $@ + +clean: + -rm -f *.o $(P) diff --git a/logpolar.cpp b/logpolar.cpp new file mode 100644 index 0000000..46b48c5 --- /dev/null +++ b/logpolar.cpp @@ -0,0 +1,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; +} |