Peter M. Groen / openbr

Commit cd5c1964df71918aced797e9deb7cffb3a10daf1

Authored by Charles Otto
1 parent 3f09f0ba

Drop pixel.cpp, it is currently unmaintained (and broken by recent changes)

Inline Side-by-side
Showing 1 changed file with 0 additions and 298 deletions
openbr/plugins/pixel.cpp deleted
View file @3f09f0b
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 "openbr_internal.h"
18   -
19   -using namespace cv;
20   -
21   -namespace br
22   -{
23   -
24   -/*!
25   - * \ingroup transforms
26   - * \brief Treat each pixel as a classification task
27   - * \author E. Taborsky \cite mmtaborsky
28   - */
29   -class PerPixelClassifierTransform : public MetaTransform
30   -{
31   - // problematic -cao
32   - Q_OBJECT
33   - Q_PROPERTY(br::Transform* transform READ get_transform WRITE set_transform RESET reset_transform)
34   - Q_PROPERTY(int pixels READ get_pixels WRITE set_pixels RESET reset_pixels STORED false)
35   - Q_PROPERTY(int orient READ get_orient WRITE set_orient RESET reset_orient STORED false)
36   - BR_PROPERTY(br::Transform*, transform, NULL)
37   - BR_PROPERTY(int, pixels, 10000)
38   - BR_PROPERTY(bool, orient, false)
39   -
40   - /*
41   - Bins:
42   - |4|3|2|
43   - |5| |1|
44   - |6|7|8|
45   - */
46   -
47   - QList<float> shift(int n, QList<float> &src) const
48   - {
49   - for (int i = 0; i < n; i++){ // Equivalent to src.append(src.takeFirst()) ?
50   - src.append(src.at(i));
51   - src.removeFirst();
52   - }
53   - return src;
54   - }
55   -
56   - void rotate(Template &src, Template &dst) const
57   - {
58   - int images = (src.m().cols)/9;
59   - dst = src;
60   - for (int i = 0; i < images; i++){
61   - double a = src.m().at<float>(7+(i*9)); //top
62   - double b = src.m().at<float>(1+(i*9)); //bottom
63   - double c = src.m().at<float>(5+(i*9)); //right
64   - double d = src.m().at<float>(3+(i*9)); //left
65   - double orientation = atan2((a-b),(c-d));
66   - int bin;
67   - if (orientation > 0){
68   - bin = ((orientation/CV_PI)*4.0 +.5);
69   - } else {
70   - bin = 8.0 + ((orientation/CV_PI)*4.0 + .5);
71   - }
72   -
73   - // put things in an order that makes sense to rotate
74   - // blugh
75   - QList<float> orderedList;
76   - QList<float> rotatedList;
77   - orderedList.insert(0, src.m().at<float>(3+(i*9)));
78   - orderedList.insert(1, src.m().at<float>(6+(i*9)));
79   - orderedList.insert(2, src.m().at<float>(7+(i*9)));
80   - orderedList.insert(3, src.m().at<float>(8+(i*9)));
81   - orderedList.insert(4, src.m().at<float>(5+(i*9)));
82   - orderedList.insert(5, src.m().at<float>(2+(i*9)));
83   - orderedList.insert(6, src.m().at<float>(1+(i*9)));
84   - orderedList.insert(7, src.m().at<float>(0+(i*9)));
85   -
86   - rotatedList = shift(bin, orderedList);
87   -
88   - dst.m().at<float>(0+(i*9)) = rotatedList.at(7);
89   - dst.m().at<float>(1+(i*9)) = rotatedList.at(6);
90   - dst.m().at<float>(2+(i*9)) = rotatedList.at(5);
91   - dst.m().at<float>(3+(i*9)) = rotatedList.at(0);
92   - dst.m().at<float>(4+(i*9)) = src.m().at<float>(4+(i*9)); // middle pixel not in orderedList
93   - dst.m().at<float>(5+(i*9)) = rotatedList.at(4);
94   - dst.m().at<float>(6+(i*9)) = rotatedList.at(1);
95   - dst.m().at<float>(7+(i*9)) = rotatedList.at(2);
96   - dst.m().at<float>(8+(i*9)) = rotatedList.at(3);
97   - }
98   - }
99   -
100   - void train(const TemplateList &trainingSet)
101   - {
102   - TemplateList pixelTemplates = TemplateList();
103   - const int length = trainingSet.length();
104   - int pixelsPerImage = pixels/length;
105   -
106   - for (int i=0; i < length; i++){
107   - Template src = trainingSet.at(i);
108   -
109   - const int mats = src.length();
110   - const int rows = src.m().rows;
111   - const int cols = src.m().cols;
112   -
113   - RNG &rng = theRNG();
114   - TemplateList srcPixelTemplates;
115   -
116   - for (int m=0; m < pixelsPerImage; m++){
117   - int index = rng.uniform(0, rows*cols);
118   - Template temp = Template(src.file, cv::Mat(1, mats, CV_32F));
119   - float *ptemp = (float*)temp.m().ptr();
120   - for (int n=0; n < mats; n++){
121   - uchar *psrc = src[n].ptr();
122   - ptemp[n] = psrc[index];
123   - }
124   - cv::Mat labelMat = src.file.value("labels").value<cv::Mat>();
125   - uchar* plabel = labelMat.ptr();
126   - temp.file.set("Label", plabel[index]);
127   -
128   - if (orient){
129   - Template rotated;
130   - rotate(temp, rotated);
131   - srcPixelTemplates.append(rotated);
132   - } else {
133   - srcPixelTemplates.append(temp);
134   - }
135   - }
136   - pixelTemplates.append(srcPixelTemplates);
137   - }
138   - transform->train(pixelTemplates);
139   - }
140   -
141   - void project(const Template &src, Template &dst) const
142   - {
143   - const int mats = src.length();
144   - const int rows = src.m().rows;
145   - const int cols = src.m().cols;
146   -
147   - dst = src; // Do we really want to copy all the src matrices into dst?
148   - dst.merge(Template(src.file, cv::Mat(src.m().rows, src.m().cols, CV_32F)));
149   - float *pdst = (float*) dst.m().ptr();
150   -
151   - for (int r = 0; r < rows; r++){
152   - for (int c = 0; c < cols; c++){
153   - Template temp = Template(src.file, cv::Mat(1, mats, CV_32F));
154   - Template dstTemp = Template(src.file, cv::Mat(1, mats, CV_32F));
155   -
156   - for (int n=0; n < mats; n++){
157   - const uchar *psrc = src[n].ptr();
158   - float *ptemp = (float*)temp[0].ptr();
159   - int index = r*cols + c;
160   - ptemp[n] = psrc[index];
161   - }
162   -
163   - if (orient){
164   - Template rotated = Template(src.file, cv::Mat(1, mats, CV_32F));
165   - rotate(temp, rotated);
166   - temp = rotated;
167   - }
168   -
169   - transform->project(temp,dstTemp);
170   - pdst[r*cols + c] = dstTemp.file.get<float>("Label");
171   - }
172   - }
173   - }
174   -};
175   -
176   -BR_REGISTER(Transform, PerPixelClassifierTransform)
177   -
178   -/*!
179   - * \ingroup transforms
180   - * \brief Construct feature vectors of neighboring pixels
181   - * \author E. Taborsky \cite mmtaborsky
182   - */
183   -class NeighborsTransform : public UntrainableMetaTransform
184   -{
185   - Q_OBJECT
186   -
187   - void project(const Template &src, Template &dst) const
188   - {
189   - int rows = src.m().rows;
190   - int cols = src.m().cols;
191   - int mats = src.length();
192   - dst.file = src.file;
193   -
194   - for (int n = 0; n < mats; n++){ //each matrix, except the last one, will be turned into 9 matrices
195   - const uchar *psrc = src[n].ptr();
196   - for (int i = -1; i < 2; i++){
197   - for (int j = -1; j < 2; j++){ // these nine matrices are shifted versions of the original
198   - cv::Mat mat = cv::Mat(rows, cols, CV_8UC1);
199   - uchar *pmat = (uchar*)mat.ptr();
200   - for (int r = 0; r < rows; r++){
201   - for (int c = 0; c < cols; c++){
202   - int index = r*cols+c;
203   - int newIndex = index + i*cols + j;
204   - if ((newIndex < 0) || (newIndex >= rows*cols)){
205   - pmat[index] = psrc[index];
206   - } else {
207   - pmat[index] = psrc[newIndex];
208   - }
209   - }
210   - }
211   - dst.push_back(mat); //add mat to dst
212   - }
213   - }
214   - }
215   - dst.push_back(src.m()); // add the last matrix
216   - }
217   -};
218   -
219   -BR_REGISTER(Transform, NeighborsTransform)
220   -
221   -/*!
222   - * \ingroup transforms
223   - * \brief To binary vector
224   - * \author E. Taborsky \cite mmtaborsky
225   - */
226   -class ToBinaryVectorTransform : public UntrainableMetaTransform
227   -{
228   - Q_OBJECT
229   - Q_PROPERTY(br::Transform* transform READ get_transform WRITE set_transform RESET reset_transform STORED false)
230   - Q_PROPERTY(int length READ get_length WRITE set_length RESET reset_length STORED false)
231   - BR_PROPERTY(br::Transform*, transform, NULL)
232   - BR_PROPERTY(int, length, -1)
233   -
234   - //needs to be updated..
235   - void project(const Template &src, Template &dst) const
236   - {
237   -
238   - dst = src;
239   - int mats = src.length();
240   - for (int i = 0; i < mats; i++){
241   - // Does this actually modify the data?
242   - dst[i]*(1.0/255.0); //scaling the input matrices to make the svm happier
243   - }
244   - for (int i = 0; i < length*(mats); i++){
245   - dst.prepend(Template(src.file, Mat::zeros(src.m().rows, src.m().cols, CV_8U)));
246   - }
247   -
248   - // original pixel values at the end
249   -
250   - Template transformed;
251   - transformed.file = src.file;
252   - transform->project(src, transformed);
253   -
254   - int rows = transformed.m().rows;
255   - int cols = transformed.m().cols;
256   -
257   - for (int i = 0; i < mats; i++){
258   - uchar *ptransformed = transformed[i].ptr();
259   - for (int r = 0; r < rows; r++){
260   - for (int c = 0; c < cols; c++){
261   - uchar index = ptransformed[r*cols+c];
262   - dst[index+(length*i)].at<uchar>(r,c) = 1;
263   - }
264   - }
265   - }
266   - }
267   -};
268   -
269   -BR_REGISTER(Transform, ToBinaryVectorTransform)
270   -
271   -/*!
272   - * \ingroup transforms
273   - * \brief If "labels" is specified, makes the last matrix into metadata
274   - * \author E. Taborsky \cite mmtaborsky
275   - */
276   -
277   -// What does this do? -cao
278   -class ToMetadataTransform : public UntrainableMetaTransform
279   -{
280   - Q_OBJECT
281   -
282   - void project(const Template &src, Template &dst) const
283   - {
284   - dst = src;
285   - if (dst.file.contains("labels")){
286   - QVariant last = qVariantFromValue(dst.m());
287   - dst.file.set("labels", last);
288   - dst.pop_back();
289   - }
290   - }
291   -
292   -};
293   -
294   -BR_REGISTER(Transform, ToMetadataTransform)
295   -
296   -} // namespace br
297   -
298   -#include "pixel.moc"