Peter M. Groen / openbr

Commit 097cfe2070f39d0f2e7be36bc907617b1dd877a1

Authored by Josh Klontz
1 parent 50f7eba5

remove multiscale

Inline Side-by-side
Showing 1 changed file with 0 additions and 199 deletions
openbr/plugins/imgproc/multiscale.cpp deleted
View file @50f7eba
1 -/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * *  
2 - * Copyright 2012 The MITRE Corporation *  
3 - * *  
4 - * Licensed under the Apache License, Version 2.0 (the "License"); *  
5 - * you may not use this file except in compliance with the License. *  
6 - * You may obtain a copy of the License at *  
7 - * *  
8 - * http://www.apache.org/licenses/LICENSE-2.0 *  
9 - * *  
10 - * Unless required by applicable law or agreed to in writing, software *  
11 - * distributed under the License is distributed on an "AS IS" BASIS, *  
12 - * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. *  
13 - * See the License for the specific language governing permissions and *  
14 - * limitations under the License. *  
15 - * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */  
16 -  
17 -#include <opencv2/imgproc/imgproc.hpp>  
18 -  
19 -#include <openbr/plugins/openbr_internal.h>  
20 -#include <openbr/core/opencvutils.h>  
21 -  
22 -using namespace cv;  
23 -  
24 -namespace br  
25 -{  
26 -  
27 -// Find avg aspect ratio  
28 -static float getAspectRatio(const TemplateList &data)  
29 -{  
30 - double tempRatio = 0;  
31 - int ratioCnt = 0;  
32 -  
33 - foreach (const Template &tmpl, data) {  
34 - QList<Rect> posRects = OpenCVUtils::toRects(tmpl.file.rects());  
35 - foreach (const Rect &posRect, posRects) {  
36 - if (posRect.x + posRect.width >= tmpl.m().cols || posRect.y + posRect.height >= tmpl.m().rows || posRect.x < 0 || posRect.y < 0) {  
37 - continue;  
38 - }  
39 - tempRatio += (float)posRect.width / (float)posRect.height;  
40 - ratioCnt += 1;  
41 - }  
42 - }  
43 - return tempRatio / (double)ratioCnt;  
44 -}  
45 -  
46 -static TemplateList cropTrainingSamples(const TemplateList &data, const float aspectRatio, const int minSize = 32, const float maxOverlap = 0.5, const int negToPosRatio = 1)  
47 -{  
48 - TemplateList result;  
49 - foreach (const Template &tmpl, data) {  
50 - QList<Rect> posRects = OpenCVUtils::toRects(tmpl.file.rects());  
51 - QList<Rect> negRects;  
52 - for (int i=0; i<posRects.size(); i++) {  
53 - Rect &posRect = posRects[i];  
54 -  
55 - // Adjust for training samples that have different aspect ratios  
56 - const int diff = int(posRect.height * aspectRatio) - posRect.width;  
57 - posRect.x -= diff / 2;  
58 - posRect.width += diff;  
59 -  
60 - // Ignore samples larger than the image  
61 - if ((posRect.x + posRect.width >= tmpl.m().cols) ||  
62 - (posRect.y + posRect.height >= tmpl.m().rows) ||  
63 - (posRect.x < 0) ||  
64 - (posRect.y < 0))  
65 - continue;  
66 -  
67 - result += Template(tmpl.file, Mat(tmpl, posRect));  
68 - result.last().file.set("Label", QString("pos"));  
69 -  
70 - // Add random negative samples  
71 - Mat m = tmpl.m();  
72 - int sample = 0;  
73 - while (sample < negToPosRatio) {  
74 - const int x = rand() % m.cols;  
75 - const int y = rand() % m.rows;  
76 - const int maxWidth = m.cols - x;  
77 - const int maxHeight = m.rows - y;  
78 - if (maxWidth <= minSize || maxHeight <= minSize)  
79 - continue;  
80 -  
81 - int height;  
82 - int width;  
83 - if (aspectRatio > (float) maxWidth / (float) maxHeight) {  
84 - width = rand() % (maxWidth - minSize) + minSize;  
85 - height = qRound(width / aspectRatio);  
86 - } else {  
87 - height = rand() % (maxHeight - minSize) + minSize;  
88 - width = qRound(height * aspectRatio);  
89 - }  
90 - Rect negRect(x, y, width, height);  
91 -  
92 - // The negative samples cannot overlap the positive samples at  
93 - // all, but they may partially overlap with other negatives.  
94 - if (OpenCVUtils::overlaps(posRects, negRect, 0) ||  
95 - OpenCVUtils::overlaps(negRects, negRect, maxOverlap))  
96 - continue;  
97 -  
98 - result += Template(tmpl.file, Mat(tmpl, negRect));  
99 - result.last().file.set("Label", QString("neg"));  
100 - sample++;  
101 - }  
102 - }  
103 - }  
104 -  
105 - return result;  
106 -}  
107 -  
108 -/*!  
109 - * \ingroup transforms  
110 - * \brief DOCUMENT ME  
111 - * \author Austin Blanton \cite imaus10  
112 - */  
113 -class BuildScalesTransform : public Transform  
114 -{  
115 - Q_OBJECT  
116 - Q_PROPERTY(br::Transform *transform READ get_transform WRITE set_transform RESET reset_transform STORED false)  
117 - Q_PROPERTY(double scaleFactor READ get_scaleFactor WRITE set_scaleFactor RESET reset_scaleFactor STORED false)  
118 - Q_PROPERTY(bool takeLargestScale READ get_takeLargestScale WRITE set_takeLargestScale RESET reset_takeLargestScale STORED false)  
119 - Q_PROPERTY(int windowWidth READ get_windowWidth WRITE set_windowWidth RESET reset_windowWidth STORED false)  
120 - Q_PROPERTY(int negToPosRatio READ get_negToPosRatio WRITE set_negToPosRatio RESET reset_negToPosRatio STORED false)  
121 - Q_PROPERTY(int minSize READ get_minSize WRITE set_minSize RESET reset_minSize STORED false)  
122 - Q_PROPERTY(double maxOverlap READ get_maxOverlap WRITE set_maxOverlap RESET reset_maxOverlap STORED false)  
123 - Q_PROPERTY(float minScale READ get_minScale WRITE set_minScale RESET reset_minScale STORED false)  
124 - BR_PROPERTY(br::Transform *, transform, NULL)  
125 - BR_PROPERTY(double, scaleFactor, 0.75)  
126 - BR_PROPERTY(bool, takeLargestScale, false)  
127 - BR_PROPERTY(int, windowWidth, 24)  
128 - BR_PROPERTY(int, negToPosRatio, 1)  
129 - BR_PROPERTY(int, minSize, 8)  
130 - BR_PROPERTY(double, maxOverlap, 0)  
131 - BR_PROPERTY(float, minScale, 1.0)  
132 -  
133 -private:  
134 - float aspectRatio;  
135 - int windowHeight;  
136 - bool skipProject;  
137 -  
138 - void train(const TemplateList &data)  
139 - {  
140 - skipProject = true;  
141 - aspectRatio = getAspectRatio(data);  
142 - windowHeight = qRound(windowWidth / aspectRatio);  
143 - if (transform->trainable) {  
144 - TemplateList full;  
145 - foreach (const Template &roi, cropTrainingSamples(data, aspectRatio, minSize, maxOverlap, negToPosRatio)) {  
146 - Mat resized;  
147 - resize(roi, resized, Size(windowWidth, windowHeight));  
148 - full += Template(roi.file, resized);  
149 - }  
150 - full.first().file.set("aspectRatio", aspectRatio);  
151 - transform->train(full);  
152 - }  
153 - }  
154 -  
155 - void project(const Template &src, Template &dst) const  
156 - {  
157 - dst = src;  
158 - if (skipProject) {  
159 - dst = src;  
160 - return;  
161 - }  
162 -  
163 - int rows = src.m().rows;  
164 - int cols = src.m().cols;  
165 - int windowHeight = (int) qRound((float) windowWidth / aspectRatio);  
166 -  
167 - float startScale;  
168 - if ((cols / rows) > aspectRatio)  
169 - startScale = qRound((float) rows / (float) windowHeight);  
170 - else  
171 - startScale = qRound((float) cols / (float) windowWidth);  
172 -  
173 - for (float scale = startScale; scale >= minScale; scale -= (1.0 - scaleFactor)) {  
174 - Template scaleImg(dst.file, Mat());  
175 - scaleImg.file.set("scale", scale);  
176 - resize(src, scaleImg, Size(qRound(cols / scale), qRound(rows / scale)));  
177 - transform->project(scaleImg, dst);  
178 - if (takeLargestScale && !dst.file.rects().empty())  
179 - return;  
180 - }  
181 - }  
182 -  
183 - void store(QDataStream &stream) const  
184 - {  
185 - transform->store(stream);  
186 - stream << aspectRatio << windowHeight;  
187 - }  
188 - void load(QDataStream &stream)  
189 - {  
190 - transform->load(stream);  
191 - stream >> aspectRatio >> windowHeight;  
192 - }  
193 -};  
194 -  
195 -BR_REGISTER(Transform, BuildScalesTransform)  
196 -  
197 -} // namespace br  
198 -  
199 -#include "imgproc/multiscale.moc"