edge.cpp 1.21 KB
Newer Older
1 2
#include "opencv2/imgproc/imgproc.hpp"
#include "opencv2/highgui/highgui.hpp"
3 4 5 6 7 8

#include <iostream>

using namespace cv;
using namespace std;

Gary Bradski's avatar
Gary Bradski committed
9 10 11 12 13 14 15 16 17
void help()
{
	cout <<
			"\nDemonstrate Canny edge detection\n"
			"Call:\n"
			"/.edge [image_name -- Default is fruits.jpg]\n" << endl;

}

18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40
int edgeThresh = 1;
Mat image, gray, edge, cedge;

// define a trackbar callback
void onTrackbar(int, void*)
{
    blur(gray, edge, Size(3,3));

    // Run the edge detector on grayscale
    Canny(edge, edge, edgeThresh, edgeThresh*3, 3);
    cedge = Scalar::all(0);
    
    image.copyTo(cedge, edge);
    imshow("Edge map", cedge);
}

int main( int argc, char** argv )
{
    char* filename = argc == 2 ? argv[1] : (char*)"fruits.jpg";

    image = imread(filename, 1);
    if(image.empty())
    {
Gary Bradski's avatar
Gary Bradski committed
41
    	help();
42 43
        return -1;
    }
Gary Bradski's avatar
Gary Bradski committed
44
    help();
45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61
    cedge.create(image.size(), image.type());
    cvtColor(image, gray, CV_BGR2GRAY);

    // Create a window
    namedWindow("Edge map", 1);

    // create a toolbar
    createTrackbar("Canny threshold", "Edge map", &edgeThresh, 100, onTrackbar);

    // Show the image
    onTrackbar(0, 0);

    // Wait for a key stroke; the same function arranges events processing
    waitKey(0);

    return 0;
}