← Back to branch summary

~paparazzi-uav/paparazzi/v5.0-manual

« back to all changes in this revision

Viewing changes to sw/ext/opencv_bebop/opencv/samples/tapi/clahe.cpp

  • Committer: Paparazzi buildbot
  • Date: 2016-05-18 15:00:29 UTC
  • Revision ID: felix.ruess+docbot@gmail.com-20160518150029-e8lgzi5kvb4p7un9
Manual import commit 4b8bbb730080dac23cf816b98908dacfabe2a8ec from v5.0 branch.

Show diffs side-by-side

added added

removed removed

Lines of Context:
sw/ext/opencv_bebop/opencv/samples/tapi/clahe.cpp
 
1
#include <iostream>
 
2
#include "opencv2/core/core.hpp"
 
3
#include "opencv2/core/ocl.hpp"
 
4
#include "opencv2/core/utility.hpp"
 
5
#include "opencv2/imgproc/imgproc.hpp"
 
6
#include "opencv2/imgcodecs.hpp"
 
7
#include "opencv2/videoio.hpp"
 
8
#include "opencv2/highgui/highgui.hpp"
 
9
 
 
10
using namespace cv;
 
11
using namespace std;
 
12
 
 
13
Ptr<CLAHE> pFilter;
 
14
int tilesize;
 
15
int cliplimit;
 
16
 
 
17
static void TSize_Callback(int pos)
 
18
{
 
19
    if(pos==0)
 
20
        pFilter->setTilesGridSize(Size(1,1));
 
21
    else
 
22
        pFilter->setTilesGridSize(Size(tilesize,tilesize));
 
23
}
 
24
 
 
25
static void Clip_Callback(int)
 
26
{
 
27
    pFilter->setClipLimit(cliplimit);
 
28
}
 
29
 
 
30
int main(int argc, char** argv)
 
31
{
 
32
    const char* keys =
 
33
        "{ i input    |                    | specify input image }"
 
34
        "{ c camera   |  0                 | specify camera id   }"
 
35
        "{ o output   | clahe_output.jpg   | specify output save path}"
 
36
        "{ h help     | false              | print help message }";
 
37
 
 
38
    cv::CommandLineParser cmd(argc, argv, keys);
 
39
    if (cmd.has("help"))
 
40
    {
 
41
        cout << "Usage : clahe [options]" << endl;
 
42
        cout << "Available options:" << endl;
 
43
        cmd.printMessage();
 
44
        return EXIT_SUCCESS;
 
45
    }
 
46
 
 
47
    string infile = cmd.get<string>("i"), outfile = cmd.get<string>("o");
 
48
    int camid = cmd.get<int>("c");
 
49
    VideoCapture capture;
 
50
 
 
51
    namedWindow("CLAHE");
 
52
    createTrackbar("Tile Size", "CLAHE", &tilesize, 32, (TrackbarCallback)TSize_Callback);
 
53
    createTrackbar("Clip Limit", "CLAHE", &cliplimit, 20, (TrackbarCallback)Clip_Callback);
 
54
 
 
55
    UMat frame, outframe;
 
56
 
 
57
    int cur_clip;
 
58
    Size cur_tilesize;
 
59
    pFilter = createCLAHE();
 
60
 
 
61
    cur_clip = (int)pFilter->getClipLimit();
 
62
    cur_tilesize = pFilter->getTilesGridSize();
 
63
    setTrackbarPos("Tile Size", "CLAHE", cur_tilesize.width);
 
64
    setTrackbarPos("Clip Limit", "CLAHE", cur_clip);
 
65
 
 
66
    if(infile != "")
 
67
    {
 
68
        imread(infile).copyTo(frame);
 
69
        if(frame.empty())
 
70
        {
 
71
            cout << "error read image: " << infile << endl;
 
72
            return EXIT_FAILURE;
 
73
        }
 
74
    }
 
75
    else
 
76
        capture.open(camid);
 
77
 
 
78
    cout << "\nControls:\n"
 
79
         << "\to - save output image\n"
 
80
         << "\tm - switch OpenCL <-> CPU mode"
 
81
         << "\tESC - exit\n";
 
82
 
 
83
    for (;;)
 
84
    {
 
85
        if(capture.isOpened())
 
86
            capture.read(frame);
 
87
        else
 
88
            imread(infile).copyTo(frame);
 
89
        if(frame.empty())
 
90
            continue;
 
91
 
 
92
        cvtColor(frame, frame, COLOR_BGR2GRAY);
 
93
        pFilter->apply(frame, outframe);
 
94
 
 
95
        imshow("CLAHE", outframe);
 
96
 
 
97
        char key = (char)waitKey(3);
 
98
        if(key == 'o')
 
99
            imwrite(outfile, outframe);
 
100
        else if(key == 27)
 
101
            break;
 
102
        else if(key == 'm')
 
103
        {
 
104
            ocl::setUseOpenCL(!cv::ocl::useOpenCL());
 
105
            cout << "Switched to " << (ocl::useOpenCL() ? "OpenCL enabled" : "CPU") << " mode\n";
 
106
        }
 
107
    }
 
108
    return EXIT_SUCCESS;
 
109
}