mirror of https://github.com/opencv/opencv.git
Open Source Computer Vision Library
https://opencv.org/
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
62 lines
1.5 KiB
62 lines
1.5 KiB
#! /usr/bin/env octave |
|
## This is a standalone program. Pass an image name as a first parameter of the program. |
|
|
|
cv; |
|
highgui; |
|
|
|
## toggle between CV_HOUGH_STANDARD and CV_HOUGH_PROBILISTIC |
|
USE_STANDARD=0; |
|
|
|
filename = "../../docs/ref/pics/building.jpg" |
|
if (size(argv, 1)>=1) |
|
filename = argv(){1}; |
|
endif |
|
|
|
src=cvLoadImage(filename, 0); |
|
if (!swig_this(src)) |
|
printf("Error opening image %s\n",filename); |
|
exit(-1); |
|
endif |
|
|
|
dst = cvCreateImage( cvGetSize(src), 8, 1 ); |
|
color_dst = cvCreateImage( cvGetSize(src), 8, 3 ); |
|
storage = cvCreateMemStorage(0); |
|
lines = 0; |
|
cvCanny( src, dst, 50, 200, 3 ); |
|
cvCvtColor( dst, color_dst, CV_GRAY2BGR ); |
|
|
|
if (USE_STANDARD) |
|
lines = cvHoughLines2( dst, storage, CV_HOUGH_STANDARD, 1, CV_PI/180, 100, 0, 0 ); |
|
|
|
for i=0:min(lines.total, 100)-1, |
|
line = lines{i}; |
|
rho = line{0}; |
|
theta = line{1}; |
|
pt1 = CvPoint(); |
|
pt2 = CvPoint(); |
|
a = cos(theta); |
|
b = sin(theta); |
|
x0 = a*rho; |
|
y0 = b*rho; |
|
pt1.x = cvRound(x0 + 1000*(-b)); |
|
pt1.y = cvRound(y0 + 1000*(a)); |
|
pt2.x = cvRound(x0 - 1000*(-b)); |
|
pt2.y = cvRound(y0 - 1000*(a)); |
|
cvLine( color_dst, pt1, pt2, CV_RGB(255,0,0), 3, 8 ); |
|
endfor |
|
|
|
else |
|
lines = cvHoughLines2( dst, storage, CV_HOUGH_PROBABILISTIC, 1, CV_PI/180, 50, 50, 10 ); |
|
for line = CvSeq_map(lines), |
|
line = line{1}; |
|
cvLine( color_dst, line{0}, line{1}, CV_RGB(255,0,0), 3, 8 ); |
|
endfor |
|
endif |
|
|
|
cvNamedWindow( "Source", 1 ); |
|
cvShowImage( "Source", src ); |
|
|
|
cvNamedWindow( "Hough", 1 ); |
|
cvShowImage( "Hough", color_dst ); |
|
|
|
cvWaitKey(0);
|
|
|