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
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
|
//! \example tutorial-contour.cpp
#include <cstdlib>
#include <iostream>
#include <visp3/core/vpConfig.h>
#include <visp3/core/vpImage.h>
#include <visp3/gui/vpDisplayFactory.h>
#include <visp3/io/vpImageIo.h>
#if defined(VISP_HAVE_MODULE_IMGPROC) && defined(VISP_HAVE_DISPLAY)
//! [Include]
#include <visp3/imgproc/vpImgproc.h>
//! [Include]
#ifdef ENABLE_VISP_NAMESPACE
using namespace VISP_NAMESPACE_NAME;
#endif
namespace
{
//! [Print contours hierarchy func]
void displayContourInfo(const VISP_NAMESPACE_NAME::vpContour &contour, int level)
{
std::cout << "\nContour:" << std::endl;
std::cout << "\tlevel: " << level << std::endl;
std::cout << "\tcontour type: " << (contour.m_contourType == VISP_NAMESPACE_NAME::CONTOUR_OUTER ? "outer contour" : "hole contour")
<< std::endl;
std::cout << "\tcontour size: " << contour.m_points.size() << std::endl;
std::cout << "\tnb children: " << contour.m_children.size() << std::endl;
for (std::vector<VISP_NAMESPACE_NAME::vpContour *>::const_iterator it = contour.m_children.begin(); it != contour.m_children.end();
++it) {
displayContourInfo(**it, level + 1);
}
}
//! [Print contours hierarchy func]
//! [Draw contours hierarchical func]
void drawContoursTree(vpImage<vpRGBa> &I, const VISP_NAMESPACE_NAME::vpContour &contour)
{
std::vector<std::vector<vpImagePoint> > contours;
contours.push_back(contour.m_points);
VISP_NAMESPACE_NAME::drawContours(I, contours, contour.m_contourType == VISP_NAMESPACE_NAME::CONTOUR_OUTER ? vpColor::red : vpColor::green);
for (std::vector<VISP_NAMESPACE_NAME::vpContour *>::const_iterator it = contour.m_children.begin(); it != contour.m_children.end();
++it) {
drawContoursTree(I, **it);
}
}
//! [Draw contours hierarchical func]
} // namespace
#endif
int main(int argc, const char **argv)
{
//! [Macro defined]
#if defined(VISP_HAVE_MODULE_IMGPROC) && defined(VISP_HAVE_DISPLAY)
//! [Macro defined]
//!
std::string input_filename = "grid36-03.pgm";
bool white_foreground = false;
VISP_NAMESPACE_NAME::vpContourRetrievalType extraction_method = VISP_NAMESPACE_NAME::CONTOUR_RETR_TREE;
for (int i = 1; i < argc; i++) {
if (std::string(argv[i]) == "--input" && i + 1 < argc) {
input_filename = std::string(argv[i + 1]);
}
else if (std::string(argv[i]) == "--white_foreground") {
white_foreground = true;
}
else if (std::string(argv[i]) == "--method" && i + 1 < argc) {
extraction_method = (VISP_NAMESPACE_NAME::vpContourRetrievalType)atoi(argv[i + 1]);
}
else if (std::string(argv[i]) == "--help" || std::string(argv[i]) == "-h") {
std::cout << "Usage: " << argv[0]
<< " [--input <input image>] [--method <0: "
"CONTOUR_RETR_TREE, 1: CONTOUR_RETR_LIST, 2: "
"CONTOUR_RETR_EXTERNAL>]"
" [--white_foreground] [--help]"
<< std::endl;
return EXIT_SUCCESS;
}
}
//! [Read]
vpImage<unsigned char> I;
vpImageIo::read(I, input_filename);
//! [Read]
vpImage<unsigned char> I_bin(I.getHeight(), I.getWidth());
vpImage<vpRGBa> I_draw_contours(I.getHeight(), I.getWidth());
#if (VISP_CXX_STANDARD >= VISP_CXX_STANDARD_11)
std::shared_ptr<vpDisplay> display = vpDisplayFactory::createDisplay(I_bin, 0, 0, "After binarisation");
std::shared_ptr<vpDisplay> display2 = vpDisplayFactory::createDisplay(I_draw_contours, I_bin.getWidth(), 10, "Contours");
#else
vpDisplay *display = vpDisplayFactory::allocateDisplay(I_bin, 0, 0, "After binarisation");
vpDisplay *display2 = vpDisplayFactory::allocateDisplay(I_draw_contours, I_bin.getWidth(), 10, "Contours");
#endif
//! [Otsu]
VISP_NAMESPACE_NAME::autoThreshold(I, VISP_NAMESPACE_NAME::AUTO_THRESHOLD_OTSU, white_foreground ? 0 : 1, white_foreground ? 1 : 0);
//! [Otsu]
for (unsigned int i = 0; i < I_bin.getSize(); i++) {
I_bin.bitmap[i] = 255 * I.bitmap[i];
}
//! [Find contours]
VISP_NAMESPACE_NAME::vpContour vp_contours;
std::vector<std::vector<vpImagePoint> > contours;
VISP_NAMESPACE_NAME::findContours(I, vp_contours, contours, extraction_method);
//! [Find contours]
//! [Draw contours]
VISP_NAMESPACE_NAME::drawContours(I_draw_contours, contours, vpColor::red);
//! [Draw contours]
vpDisplay::display(I_bin);
vpDisplay::display(I_draw_contours);
vpDisplay::displayText(I_draw_contours, 20, 20, "Click to draw outer / hole contours.", vpColor::red);
vpDisplay::flush(I_bin);
vpDisplay::flush(I_draw_contours);
vpDisplay::getClick(I_draw_contours);
I_draw_contours = vpRGBa(0);
//! [Draw contours hierarchical]
drawContoursTree(I_draw_contours, vp_contours);
//! [Draw contours hierarchical]
displayContourInfo(vp_contours, 0);
vpDisplay::display(I_bin);
vpDisplay::display(I_draw_contours);
vpDisplay::displayText(I_draw_contours, 20, 20, "Click to quit.", vpColor::red);
vpDisplay::displayText(I_draw_contours, 20, I_draw_contours.getWidth() - 200, "Outer contour", vpColor::red);
vpDisplay::displayText(I_draw_contours, 20, I_draw_contours.getWidth() - 100, "Hole contour", vpColor::green);
vpDisplay::flush(I_bin);
vpDisplay::flush(I_draw_contours);
vpDisplay::getClick(I_draw_contours);
#if (VISP_CXX_STANDARD < VISP_CXX_STANDARD_11) && defined(VISP_HAVE_DISPLAY)
if (display != nullptr) {
delete display;
}
if (display2 != nullptr) {
delete display2;
}
#endif
#else
(void)argc;
(void)argv;
#endif
return EXIT_SUCCESS;
}
|
