Peter M. Groen / openbr

Commit a50d515a8c15222376909c1a76e9d16b3a318bdf

Authored by Josh Klontz
1 parent 800affe8

remove procustes

Inline Side-by-side
Showing 1 changed file with 0 additions and 155 deletions
openbr/plugins/metadata/procrustes.cpp deleted
View file @800affe
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 <Eigen/Dense>  
18 -  
19 -#include <openbr/plugins/openbr_internal.h>  
20 -#include <openbr/core/opencvutils.h>  
21 -#include <openbr/core/eigenutils.h>  
22 -  
23 -using namespace cv;  
24 -  
25 -namespace br  
26 -{  
27 -  
28 -/*!  
29 - * \ingroup transforms  
30 - * \brief Procrustes alignment of points  
31 - * \author Scott Klum \cite sklum  
32 - */  
33 -class ProcrustesTransform : public MetadataTransform  
34 -{  
35 - Q_OBJECT  
36 -  
37 - Q_PROPERTY(bool warp READ get_warp WRITE set_warp RESET reset_warp STORED false)  
38 - BR_PROPERTY(bool, warp, true)  
39 -  
40 - Eigen::MatrixXf meanShape;  
41 -  
42 - void train(const TemplateList &data)  
43 - {  
44 - QList< QList<QPointF> > normalizedPoints;  
45 -  
46 - // Normalize all sets of points  
47 - foreach (br::Template datum, data) {  
48 - QList<QPointF> points = datum.file.points();  
49 - QList<QRectF> rects = datum.file.rects();  
50 -  
51 - if (points.empty() || rects.empty()) continue;  
52 -  
53 - // Assume rect appended last was bounding box  
54 - points.append(rects.last().topLeft());  
55 - points.append(rects.last().topRight());  
56 - points.append(rects.last().bottomLeft());  
57 - points.append(rects.last().bottomRight());  
58 -  
59 - // Center shape at origin  
60 - Scalar mean = cv::mean(OpenCVUtils::toPoints(points).toVector().toStdVector());  
61 - for (int i = 0; i < points.size(); i++) points[i] -= QPointF(mean[0],mean[1]);  
62 -  
63 - // Remove scale component  
64 - float norm = cv::norm(OpenCVUtils::toPoints(points).toVector().toStdVector());  
65 - for (int i = 0; i < points.size(); i++) points[i] /= norm;  
66 -  
67 - normalizedPoints.append(points);  
68 - }  
69 -  
70 - if (normalizedPoints.empty()) qFatal("Unable to calculate normalized points");  
71 -  
72 - // Determine mean shape, assuming all shapes contain the same number of points  
73 - meanShape = Eigen::MatrixXf(normalizedPoints[0].size(), 2);  
74 -  
75 - for (int i = 0; i < normalizedPoints[0].size(); i++) {  
76 - double x = 0;  
77 - double y = 0;  
78 -  
79 - for (int j = 0; j < normalizedPoints.size(); j++) {  
80 - x += normalizedPoints[j][i].x();  
81 - y += normalizedPoints[j][i].y();  
82 - }  
83 -  
84 - x /= (double)normalizedPoints.size();  
85 - y /= (double)normalizedPoints.size();  
86 -  
87 - meanShape(i,0) = x;  
88 - meanShape(i,1) = y;  
89 - }  
90 - }  
91 -  
92 - void projectMetadata(const File &src, File &dst) const  
93 - {  
94 - QList<QPointF> points = src.points();  
95 - QList<QRectF> rects = src.rects();  
96 -  
97 - if (points.empty() || rects.empty()) {  
98 - dst = src;  
99 - if (Globals->verbose) qWarning("Procrustes alignment failed because points or rects are empty.");  
100 - return;  
101 - }  
102 -  
103 - // Assume rect appended last was bounding box  
104 - points.append(rects.last().topLeft());  
105 - points.append(rects.last().topRight());  
106 - points.append(rects.last().bottomLeft());  
107 - points.append(rects.last().bottomRight());  
108 -  
109 - Scalar mean = cv::mean(OpenCVUtils::toPoints(points).toVector().toStdVector());  
110 - for (int i = 0; i < points.size(); i++) points[i] -= QPointF(mean[0],mean[1]);  
111 -  
112 - Eigen::MatrixXf srcMat(points.size(), 2);  
113 - float norm = cv::norm(OpenCVUtils::toPoints(points).toVector().toStdVector());  
114 - for (int i = 0; i < points.size(); i++) {  
115 - points[i] /= norm;  
116 - srcMat(i,0) = points[i].x();  
117 - srcMat(i,1) = points[i].y();  
118 - }  
119 -  
120 - Eigen::JacobiSVD<Eigen::MatrixXf> svd(srcMat.transpose()*meanShape, Eigen::ComputeThinU | Eigen::ComputeThinV);  
121 - Eigen::MatrixXf R = svd.matrixU()*svd.matrixV().transpose();  
122 -  
123 - dst = src;  
124 -  
125 - // Store procrustes stats in the order:  
126 - // R(0,0), R(1,0), R(1,1), R(0,1), mean_x, mean_y, norm  
127 - QList<float> procrustesStats;  
128 - procrustesStats << R(0,0) << R(1,0) << R(1,1) << R(0,1) << mean[0] << mean[1] << norm;  
129 - dst.setList<float>("ProcrustesStats",procrustesStats);  
130 -  
131 - if (warp) {  
132 - Eigen::MatrixXf dstMat = srcMat*R;  
133 - for (int i = 0; i < dstMat.rows(); i++) {  
134 - dst.appendPoint(QPointF(dstMat(i,0),dstMat(i,1)));  
135 - }  
136 - }  
137 - }  
138 -  
139 - void store(QDataStream &stream) const  
140 - {  
141 - stream << meanShape;  
142 - }  
143 -  
144 - void load(QDataStream &stream)  
145 - {  
146 - stream >> meanShape;  
147 - }  
148 -  
149 -};  
150 -  
151 -BR_REGISTER(Transform, ProcrustesTransform)  
152 -  
153 -} // namespace br  
154 -  
155 -#include "metadata/procrustes.moc"