Commit 05383a8c19f4e2d94f9d9bb9a1318984bb9d2b07

Authored by Josh Klontz
1 parent b0487904

code cleanup

Showing 1 changed file with 18 additions and 18 deletions
sdk/plugins/meta.cpp
@@ -26,13 +26,13 @@ using namespace cv; @@ -26,13 +26,13 @@ using namespace cv;
26 namespace br 26 namespace br
27 { 27 {
28 28
29 -static TemplateList Simplified(const TemplateList &templates) 29 +static TemplateList Expanded(const TemplateList &templates)
30 { 30 {
31 - TemplateList simplified; 31 + TemplateList expanded;
32 foreach (const Template &t, templates) { 32 foreach (const Template &t, templates) {
33 if (t.isEmpty()) { 33 if (t.isEmpty()) {
34 if (!t.file.getBool("enrollAll")) 34 if (!t.file.getBool("enrollAll"))
35 - simplified.append(t); 35 + expanded.append(t);
36 continue; 36 continue;
37 } 37 }
38 38
@@ -46,13 +46,13 @@ static TemplateList Simplified(const TemplateList &templates) @@ -46,13 +46,13 @@ static TemplateList Simplified(const TemplateList &templates)
46 46
47 for (int i=0; i<t.size(); i++) { 47 for (int i=0; i<t.size(); i++) {
48 if (!fte || !t.file.getBool("enrollAll")) { 48 if (!fte || !t.file.getBool("enrollAll")) {
49 - simplified.append(Template(t.file, t[i]));  
50 - simplified.last().file.setROIs(ROIs.mid(i*ROIStep, ROIStep));  
51 - simplified.last().file.setLandmarks(landmarks.mid(i*landmarkStep, landmarkStep)); 49 + expanded.append(Template(t.file, t[i]));
  50 + expanded.last().file.setROIs(ROIs.mid(i*ROIStep, ROIStep));
  51 + expanded.last().file.setLandmarks(landmarks.mid(i*landmarkStep, landmarkStep));
52 } 52 }
53 } 53 }
54 } 54 }
55 - return simplified; 55 + return expanded;
56 } 56 }
57 57
58 static void _train(Transform *transform, const TemplateList *data) 58 static void _train(Transform *transform, const TemplateList *data)
@@ -93,6 +93,7 @@ static void incrementStep() @@ -93,6 +93,7 @@ static void incrementStep()
93 * The source br::Template is given to the first transform and the resulting br::Template is passed to the next transform, etc. 93 * The source br::Template is given to the first transform and the resulting br::Template is passed to the next transform, etc.
94 * 94 *
95 * \see ExpandTransform 95 * \see ExpandTransform
  96 + * \see ForkTransform
96 */ 97 */
97 class PipeTransform : public MetaTransform 98 class PipeTransform : public MetaTransform
98 { 99 {
@@ -142,10 +143,9 @@ class PipeTransform : public MetaTransform @@ -142,10 +143,9 @@ class PipeTransform : public MetaTransform
142 void backProject(const Template &dst, Template &src) const 143 void backProject(const Template &dst, Template &src) const
143 { 144 {
144 src = dst; 145 src = dst;
145 - //reverse order in which transforms are processed 146 + // Reverse order in which transforms are processed
146 int length = transforms.length(); 147 int length = transforms.length();
147 - //foreach (const Transform *f, transforms) {  
148 - for (int i=length-1; i>=0; i--){ 148 + for (int i=length-1; i>=0; i--) {
149 Transform *f = transforms.at(i); 149 Transform *f = transforms.at(i);
150 try { 150 try {
151 src >> *f; 151 src >> *f;
@@ -156,18 +156,17 @@ class PipeTransform : public MetaTransform @@ -156,18 +156,17 @@ class PipeTransform : public MetaTransform
156 } 156 }
157 } 157 }
158 } 158 }
159 -  
160 }; 159 };
161 160
162 BR_REGISTER(Transform, PipeTransform) 161 BR_REGISTER(Transform, PipeTransform)
163 162
164 /*! 163 /*!
165 * \ingroup transforms 164 * \ingroup transforms
166 - * \brief Transforms in series. 165 + * \brief Transforms in series with expansion step.
167 * \author Josh Klontz \cite jklontz 166 * \author Josh Klontz \cite jklontz
168 * 167 *
169 * The source br::Template is given to the first transform and the resulting br::Template is passed to the next transform, etc. 168 * The source br::Template is given to the first transform and the resulting br::Template is passed to the next transform, etc.
170 - * Each matrix is reorganized into a new template before continuing. 169 + * Each matrix is expanded into its own template between steps.
171 * 170 *
172 * \see PipeTransform 171 * \see PipeTransform
173 */ 172 */
@@ -185,7 +184,7 @@ class ExpandTransform : public MetaTransform @@ -185,7 +184,7 @@ class ExpandTransform : public MetaTransform
185 for (int i=0; i<transforms.size(); i++) { 184 for (int i=0; i<transforms.size(); i++) {
186 transforms[i]->train(copy); 185 transforms[i]->train(copy);
187 copy >> *transforms[i]; 186 copy >> *transforms[i];
188 - copy = Simplified(copy); 187 + copy = Expanded(copy);
189 incrementStep(); 188 incrementStep();
190 } 189 }
191 190
@@ -211,17 +210,16 @@ class ExpandTransform : public MetaTransform @@ -211,17 +210,16 @@ class ExpandTransform : public MetaTransform
211 dst = src; 210 dst = src;
212 for (int i=0; i<transforms.size(); i++) { 211 for (int i=0; i<transforms.size(); i++) {
213 dst >> *transforms[i]; 212 dst >> *transforms[i];
214 - dst = Simplified(dst); 213 + dst = Expanded(dst);
215 } 214 }
216 } 215 }
217 216
218 void backProject(const Template &dst, Template &src) const 217 void backProject(const Template &dst, Template &src) const
219 { 218 {
220 src = dst; 219 src = dst;
221 - //reverse order in which transforms are processed 220 + // Reverse order in which transforms are processed
222 int length = transforms.length(); 221 int length = transforms.length();
223 - //foreach (const Transform *f, transforms) {  
224 - for (int i=length-1; i>=0; i--){ 222 + for (int i=length-1; i>=0; i--) {
225 Transform *f = transforms.at(i); 223 Transform *f = transforms.at(i);
226 try { 224 try {
227 src >> *f; 225 src >> *f;
@@ -242,6 +240,8 @@ BR_REGISTER(Transform, ExpandTransform) @@ -242,6 +240,8 @@ BR_REGISTER(Transform, ExpandTransform)
242 * \author Josh Klontz \cite jklontz 240 * \author Josh Klontz \cite jklontz
243 * 241 *
244 * The source br::Template is seperately given to each transform and the results are appended together. 242 * The source br::Template is seperately given to each transform and the results are appended together.
  243 + *
  244 + * \see PipeTransform
245 */ 245 */
246 class ForkTransform : public MetaTransform 246 class ForkTransform : public MetaTransform
247 { 247 {