Peter M. Groen / openbr

Commit 74d09154611152fcf768fa182fe03e1f0a41fcc6

Authored by Josh Klontz
1 parent c0232698

removed pp5 plugin

Inline Side-by-side
Showing 2 changed files with 0 additions and 619 deletions
openbr/plugins/classification/pp5.cpp deleted
View file @c023269
1 -#include <QDebug>  
2 -#include <QFileInfo>  
3 -#include <QMap>  
4 -#include <QMutex>  
5 -#include <QSharedPointer>  
6 -#include <QString>  
7 -#include <QStringList>  
8 -#include <QThreadPool>  
9 -#include <QVariant>  
10 -#include <pittpatt_errors.h>  
11 -#include <pittpatt_raw_image_io.h>  
12 -#include <pittpatt_sdk.h>  
13 -#include <pittpatt_license.h>  
14 -#include "openbr/plugins/openbr_internal.h"  
15 -#include "openbr/core/resource.h"  
16 -  
17 -#define TRY(CC) \  
18 -{ \  
19 - if ((CC) != PPR_SUCCESS) qFatal("%d error (%s, %d): %s.", CC, __FILE__, __LINE__, ppr_error_message(CC)); \  
20 -}  
21 -  
22 -#define TRY_VIDEO(CC) \  
23 -{ \  
24 - if ((CC) != PPR_VIDEO_IO_SUCCESS) qFatal("%d error (%s, %d): %s.", CC, __FILE__, __LINE__, ppr_video_io_error_message(CC)); \  
25 -}  
26 -  
27 -#define TRY_RAW_IMAGE(CC) \  
28 -{ \  
29 - if ((CC) != PPR_RAW_IMAGE_SUCCESS) qFatal("%d error (%s, %d): %s.", CC, __FILE__, __LINE__, ppr_raw_image_error_message(CC)); \  
30 -}  
31 -  
32 -using namespace br;  
33 -  
34 -/*!  
35 - * \ingroup initializers  
36 - * \brief Initialize PP5  
37 - * \author Josh Klontz \cite jklontz  
38 - * \author E. Taborsky \cite mmtaborsky  
39 - * \warning PittPatt 5.x.x is known to NOT work with MinGW-w64 due to a segfault in ppr_initialize_sdk.  
40 - */  
41 -class PP5Initializer : public Initializer  
42 -{  
43 - Q_OBJECT  
44 -  
45 - void initialize() const  
46 - {  
47 - TRY(ppr_initialize_sdk(qPrintable(Globals->sdkPath + "/share/openbr/models/pp5/"), my_license_id, my_license_key))  
48 - Globals->abbreviations.insert("PP5","Open+Expand+PP5Enroll!PP5Gallery");  
49 - Globals->abbreviations.insert("PP5Register", "PP5Enroll(true,true,0.02,5,Extended)+RenameFirst([eyeL,PP5_Landmark0_Right_Eye],Affine_0)+RenameFirst([eyeR,PP5_Landmark1_Left_Eye],Affine_1)");  
50 - Globals->abbreviations.insert("PP5CropFace", "Open+PP5Enroll(true)+RenameFirst([eyeL,PP5_Landmark0_Right_Eye],Affine_0)+RenameFirst([eyeR,PP5_Landmark1_Left_Eye],Affine_1)+Affine(128,128,0.25,0.35)+Cvt(Gray)");  
51 - }  
52 -  
53 - void finalize() const  
54 - {  
55 - ppr_finalize_sdk();  
56 - }  
57 -};  
58 -  
59 -BR_REGISTER(Initializer, PP5Initializer)  
60 -  
61 -/*!  
62 - * \brief PP5 context  
63 - * \author Josh Klontz \cite jklontz  
64 - * \author E. Taborsky \cite mmtaborsky  
65 - */  
66 -struct PP5Context  
67 -{  
68 - ppr_context_type context;  
69 -  
70 - PP5Context(bool detectOnly = false, float adaptiveMinSize = 0.01f, int minSize = 4, ppr_landmark_range_type landmarkRange = PPR_LANDMARK_RANGE_COMPREHENSIVE, int searchPruningAggressiveness = 0)  
71 - {  
72 - ppr_settings_type default_settings = ppr_get_default_settings();  
73 -  
74 - default_settings.detection.adaptive_max_size = 1.f;  
75 - default_settings.detection.adaptive_min_size = adaptiveMinSize;  
76 - default_settings.detection.detect_best_face_only = !Globals->enrollAll;  
77 - default_settings.detection.enable = 1;  
78 - default_settings.detection.min_size = minSize;  
79 - default_settings.detection.search_pruning_aggressiveness = searchPruningAggressiveness;  
80 - default_settings.detection.use_serial_face_detection = 1;  
81 -  
82 - default_settings.landmarks.enable = 1;  
83 - default_settings.landmarks.landmark_range = landmarkRange;  
84 - default_settings.landmarks.manually_detect_landmarks = 0;  
85 -  
86 - default_settings.recognition.automatically_extract_templates = !detectOnly;  
87 - default_settings.recognition.enable_comparison = !detectOnly;  
88 - default_settings.recognition.enable_extraction = !detectOnly;  
89 - default_settings.recognition.num_comparison_threads = 1;  
90 - default_settings.recognition.recognizer = PPR_RECOGNIZER_MULTI_POSE;  
91 - TRY(ppr_initialize_context(default_settings, &context))  
92 - }  
93 -  
94 - ~PP5Context()  
95 - {  
96 - TRY(ppr_finalize_context(context))  
97 - }  
98 -  
99 - static void createRawImage(const cv::Mat &src, ppr_raw_image_type &dst)  
100 - {  
101 - if (!src.isContinuous()) qFatal("PP5Context::createRawImage requires continuous data.");  
102 - else if (src.channels() == 3) ppr_raw_image_create(&dst, src.cols, src.rows, PPR_RAW_IMAGE_BGR24);  
103 - else if (src.channels() == 1) ppr_raw_image_create(&dst, src.cols, src.rows, PPR_RAW_IMAGE_GRAY8);  
104 - else qFatal("PP5Context::createRawImage invalid channel count.");  
105 - memcpy(dst.data, src.data, src.channels()*src.rows*src.cols);  
106 - }  
107 -  
108 - void createMat(const ppr_face_type &src, cv::Mat &dst) const  
109 - {  
110 - ppr_flat_data_type flat_data;  
111 - TRY(ppr_flatten_face(context,src,&flat_data))  
112 - dst = cv::Mat(1, flat_data.length, CV_8UC1, flat_data.data).clone();  
113 - ppr_free_flat_data(flat_data);  
114 - }  
115 -  
116 - void createFace(const cv::Mat &src, ppr_face_type *dst) const  
117 - {  
118 - ppr_flat_data_type flat_data;  
119 - flat_data.length = src.cols;  
120 - flat_data.data = src.data;  
121 - TRY(ppr_unflatten_face(context, flat_data, dst))  
122 - }  
123 -  
124 - static QString toString(const ppr_landmark_category_type &category)  
125 - {  
126 - switch (category) {  
127 - case PPR_LANDMARK_CATEGORY_LEFT_EYE:  
128 - return "Left_Eye";  
129 - case PPR_LANDMARK_CATEGORY_RIGHT_EYE:  
130 - return "Right_Eye";  
131 - case PPR_LANDMARK_CATEGORY_NOSE_BASE:  
132 - return "Nose_Base";  
133 - case PPR_LANDMARK_CATEGORY_NOSE_BRIDGE:  
134 - return "Nose_Bridge";  
135 - case PPR_LANDMARK_CATEGORY_EYE_NOSE:  
136 - return "Eye_Nose";  
137 - case PPR_LANDMARK_CATEGORY_LEFT_UPPER_CHEEK:  
138 - return "Left_Upper_Cheek";  
139 - case PPR_LANDMARK_CATEGORY_LEFT_LOWER_CHEEK:  
140 - return "Left_Lower_Cheek";  
141 - case PPR_LANDMARK_CATEGORY_RIGHT_UPPER_CHEEK:  
142 - return "Right_Upper_Cheek";  
143 - case PPR_LANDMARK_CATEGORY_RIGHT_LOWER_CHEEK:  
144 - return "Right_Lower_Cheek";  
145 - case PPR_NUM_LANDMARK_CATEGORIES:  
146 - return "Num_Landmark_Categories";  
147 - }  
148 -  
149 - return "Unknown";  
150 - }  
151 -  
152 - static QMap<QString,QVariant> toMetadata(const ppr_face_type &face)  
153 - {  
154 - QMap<QString,QVariant> metadata;  
155 -  
156 - ppr_face_attributes_type face_attributes;  
157 - ppr_get_face_attributes(face, &face_attributes);  
158 - metadata.insert("FrontalFace", QRectF(face_attributes.position.x - face_attributes.dimensions.width/2,  
159 - face_attributes.position.y - face_attributes.dimensions.height/2,  
160 - face_attributes.dimensions.width,  
161 - face_attributes.dimensions.height));  
162 - metadata.insert("Confidence", face_attributes.confidence);  
163 - metadata.insert("PP5_Face_Roll", face_attributes.rotation.roll);  
164 - metadata.insert("PP5_Face_Pitch", face_attributes.rotation.pitch);  
165 - metadata.insert("PP5_Face_Yaw", face_attributes.rotation.yaw);  
166 - metadata.insert("PP5_Face_HasThumbnail", face_attributes.has_thumbnail);  
167 - metadata.insert("PP5_Face_NumLandmarks", face_attributes.num_landmarks);  
168 - metadata.insert("PP5_Face_Size", face_attributes.size);  
169 - metadata.insert("PP5_TrackingInfo_ConfidenceLevel", face_attributes.tracking_info.confidence_level);  
170 - metadata.insert("PP5_TrackingInfo_FrameNumber", face_attributes.tracking_info.frame_number);  
171 - metadata.insert("PP5_TrackingInfo_TrackID", face_attributes.tracking_info.track_id);  
172 -  
173 - ppr_landmark_list_type landmark_list;  
174 - TRY(ppr_get_face_landmarks(face, &landmark_list))  
175 -  
176 - QList<ppr_landmark_category_type> categories;  
177 - categories << PPR_LANDMARK_CATEGORY_RIGHT_EYE  
178 - << PPR_LANDMARK_CATEGORY_LEFT_EYE  
179 - << PPR_LANDMARK_CATEGORY_NOSE_BASE  
180 - << PPR_LANDMARK_CATEGORY_NOSE_BRIDGE  
181 - << PPR_LANDMARK_CATEGORY_EYE_NOSE  
182 - << PPR_LANDMARK_CATEGORY_LEFT_UPPER_CHEEK  
183 - << PPR_LANDMARK_CATEGORY_LEFT_LOWER_CHEEK  
184 - << PPR_LANDMARK_CATEGORY_RIGHT_UPPER_CHEEK  
185 - << PPR_LANDMARK_CATEGORY_RIGHT_LOWER_CHEEK;  
186 - for (int i=0; i<categories.size(); i++) {  
187 - ppr_landmark_category_type category = categories[i];  
188 - QString metadataString = QString("PP5_Landmark%1_%2").arg(QString::number(i), toString(category));  
189 -  
190 - bool found = false;  
191 - for (int j=0; j<landmark_list.length; j++) {  
192 - ppr_landmark_type &landmark = landmark_list.landmarks[j];  
193 - if (landmark.category != category) continue;  
194 -  
195 - metadata.insert(metadataString, QPointF(landmark.position.x, landmark.position.y));  
196 - found = true;  
197 - break;  
198 - }  
199 -  
200 - if (!found) {  
201 - metadata.insert(metadataString, QPointF(std::numeric_limits<float>::quiet_NaN(), std::numeric_limits<float>::quiet_NaN()));  
202 - }  
203 - }  
204 -  
205 - ppr_free_landmark_list(landmark_list);  
206 -  
207 - return metadata;  
208 - }  
209 -  
210 - void compareNative(ppr_gallery_type target, const QList<int> &targetIDs, ppr_gallery_type query, const QList<int> &queryIDs, Output *output) const  
211 - {  
212 - ppr_similarity_matrix_type simmat;  
213 - TRY(ppr_compare_galleries(context, query, target, &simmat))  
214 - for (int i=0; i<queryIDs.size(); i++) {  
215 - int query_subject_id = queryIDs[i];  
216 - for (int j=0; j<targetIDs.size(); j++) {  
217 - int target_subject_id = targetIDs[j];  
218 - float score = -std::numeric_limits<float>::max();  
219 - if ((query_subject_id != -1) && (target_subject_id != -1)) {  
220 - TRY(ppr_get_subject_similarity_score(context, simmat, query_subject_id, target_subject_id, &score))  
221 - }  
222 - output->setRelative(score, i, j);  
223 - }  
224 - }  
225 - ppr_free_similarity_matrix(simmat);  
226 - }  
227 -  
228 - void enroll(const TemplateList &templates, ppr_gallery_type *gallery, QList<int> &subject_ids) const  
229 - {  
230 - int subject_id = 0, face_id = 0;  
231 - foreach (const Template &src, templates) {  
232 - if (!src.empty() && src.m().data) {  
233 - foreach (const cv::Mat &m, src) {  
234 - ppr_face_type face;  
235 - createFace(m, &face);  
236 - TRY(ppr_add_face(context, gallery, face, subject_id, face_id))  
237 - face_id++;  
238 - ppr_free_face(face);  
239 - }  
240 - subject_ids.append(subject_id);  
241 - subject_id++;  
242 - } else {  
243 - subject_ids.append(-1);  
244 - }  
245 - }  
246 - }  
247 -  
248 -};  
249 -  
250 -/*!  
251 - * \ingroup transforms  
252 - * \brief Enroll faces in PP5  
253 - * \author Josh Klontz \cite jklontz  
254 - * \author E. Taborsky \cite mmtaborsky  
255 - * \br_property bool detectOnly If true, enroll all detected faces. Otherwise, only enroll faces suitable for recognition. Default is false.  
256 - * \br_property bool requireLandmarks If true, require the right eye, left eye, and nose base to be detectable by PP5. If this does not happen FTE is set to true for that template. Default is false.  
257 - * \br_property float adaptiveMinSize The minimum face size as a percentage of total image width. 0.1 corresponds to a minimum face size of 10% the total image width. Default is 0.01.  
258 - * \br_property int minSize The absolute minimum face size to search for. This is not a pixel value. Please see PittPatt documentation for the relationship between minSize and pixel IPD. Default is 4.  
259 - * \br_property enum landmarkRange Range of landmarks to search for. Options are Frontal, Extended, Full, and Comprehensive. Default is Comprehensive.  
260 - * \br_property int searchPruningAggressiveness The amount of aggressiveness involved in search for faces in images. 0 means all scales and locations are searched. 1 means fewer detectors are used in the early stages but all scales are still searched. 2-4 means that the largest faces are found first and then fewer scales are searched. Default is 0.  
261 - */  
262 -class PP5EnrollTransform : public UntrainableMetaTransform  
263 -{  
264 - Q_OBJECT  
265 - Q_PROPERTY(bool detectOnly READ get_detectOnly WRITE set_detectOnly RESET reset_detectOnly STORED false)  
266 - Q_PROPERTY(bool requireLandmarks READ get_requireLandmarks WRITE set_requireLandmarks RESET reset_requireLandmarks STORED false)  
267 - Q_PROPERTY(float adaptiveMinSize READ get_adaptiveMinSize WRITE set_adaptiveMinSize RESET reset_adaptiveMinSize STORED false)  
268 - Q_PROPERTY(int minSize READ get_minSize WRITE set_minSize RESET reset_minSize STORED false)  
269 - Q_PROPERTY(LandmarkRange landmarkRange READ get_landmarkRange WRITE set_landmarkRange RESET reset_landmarkRange STORED false)  
270 - Q_PROPERTY(int searchPruningAggressiveness READ get_searchPruningAggressiveness WRITE set_searchPruningAggressiveness RESET reset_searchPruningAggressiveness STORED false)  
271 -  
272 -public:  
273 - enum LandmarkRange  
274 - {  
275 - Frontal = PPR_LANDMARK_RANGE_FRONTAL,  
276 - Extended = PPR_LANDMARK_RANGE_EXTENDED,  
277 - Full = PPR_LANDMARK_RANGE_FULL,  
278 - Comprehensive = PPR_LANDMARK_RANGE_COMPREHENSIVE  
279 - };  
280 - Q_ENUMS(LandmarkRange)  
281 -  
282 -private:  
283 - BR_PROPERTY(bool, detectOnly, false)  
284 - BR_PROPERTY(bool, requireLandmarks, false)  
285 - BR_PROPERTY(float, adaptiveMinSize, 0.01f)  
286 - BR_PROPERTY(int, minSize, 4)  
287 - BR_PROPERTY(LandmarkRange, landmarkRange, Comprehensive)  
288 - BR_PROPERTY(int, searchPruningAggressiveness, 0)  
289 -  
290 - Resource<PP5Context> contexts;  
291 -  
292 - struct PP5ContextMaker : public ResourceMaker<PP5Context>  
293 - {  
294 - PP5ContextMaker(PP5EnrollTransform *pp5EnrollTransform)  
295 - : pp5EnrollTransform(pp5EnrollTransform) {}  
296 -  
297 - private:  
298 - PP5EnrollTransform *pp5EnrollTransform;  
299 -  
300 - PP5Context *make() const  
301 - {  
302 - return new PP5Context(pp5EnrollTransform->detectOnly,  
303 - pp5EnrollTransform->adaptiveMinSize,  
304 - pp5EnrollTransform->minSize,  
305 - (ppr_landmark_range_type) pp5EnrollTransform->landmarkRange,  
306 - pp5EnrollTransform->searchPruningAggressiveness);  
307 - }  
308 - };  
309 -  
310 - void init()  
311 - {  
312 - contexts.setResourceMaker(new PP5ContextMaker(this));  
313 - }  
314 -  
315 - void project(const Template &src, Template &dst) const  
316 - {  
317 - if (Globals->enrollAll)  
318 - qFatal("single template project doesn't support enrollAll");  
319 -  
320 - TemplateList srcList;  
321 - srcList.append(src);  
322 - TemplateList dstList;  
323 - project(srcList, dstList);  
324 - dst = dstList.first();  
325 - }  
326 -  
327 - void project(const TemplateList &srcList, TemplateList &dstList) const  
328 - {  
329 - // Nothing to do here  
330 - if (srcList.empty())  
331 - return;  
332 -  
333 - PP5Context *context = contexts.acquire();  
334 -  
335 - foreach (const Template &src, srcList) {  
336 - bool foundFace = false;  
337 - if (!src.isEmpty()) {  
338 - ppr_raw_image_type raw_image;  
339 - PP5Context::createRawImage(src, raw_image);  
340 - ppr_image_type image;  
341 - TRY(ppr_create_image(raw_image, &image))  
342 - ppr_face_list_type face_list;  
343 - TRY(ppr_detect_faces(context->context, image, &face_list))  
344 -  
345 - for (int i=0; i<face_list.length; i++) {  
346 - ppr_face_type face = face_list.faces[i];  
347 - if (!detectOnly) {  
348 - int extractable;  
349 - TRY(ppr_is_template_extractable(context->context, face, &extractable))  
350 - if (!extractable)  
351 - continue;  
352 - }  
353 - foundFace = true;  
354 -  
355 - cv::Mat m;  
356 - if (detectOnly) {  
357 - m = src;  
358 - } else {  
359 - TRY(ppr_extract_face_template(context->context, image, &face))  
360 - context->createMat(face, m);  
361 - }  
362 - Template dst;  
363 - dst.file = src.file;  
364 -  
365 - dst.file.append(PP5Context::toMetadata(face));  
366 - if (requireLandmarks) {  
367 - QPointF right = dst.file.get<QPointF>("PP5_Landmark0_Right_Eye");  
368 - QPointF left = dst.file.get<QPointF>("PP5_Landmark1_Left_Eye");  
369 - QPointF nose = dst.file.get<QPointF>("PP5_Landmark2_Nose_Base");  
370 - // a number not equaling itself means it's NaN  
371 - // there should be no NaNs for the 3 special landmarks  
372 - if (dst.file.get<int>("PP5_Face_NumLandmarks") < 3 ||  
373 - right.x() != right.x() || right.y() != right.y() ||  
374 - left.x() != left.x() || left.y() != left.y() ||  
375 - nose.x() != nose.x() || nose.y() != nose.y())  
376 - {  
377 - dst.file.fte = true;  
378 - }  
379 - }  
380 - dst += m;  
381 - dstList.append(dst);  
382 -  
383 - // Found a face, nothing else to do (if we aren't trying to find multiple faces).  
384 - if (!Globals->enrollAll)  
385 - break;  
386 - }  
387 -  
388 - ppr_free_face_list(face_list);  
389 - ppr_free_image(image);  
390 - ppr_raw_image_free(raw_image);  
391 - }  
392 -  
393 - // No faces were detected when we were expecting one, output something with FTE set.  
394 - if (!foundFace && !Globals->enrollAll) {  
395 - dstList.append(Template(src.file, detectOnly ? src.m() : cv::Mat()));  
396 - dstList.last().file.fte = true;  
397 - }  
398 - }  
399 -  
400 - contexts.release(context);  
401 - }  
402 -};  
403 -  
404 -BR_REGISTER(Transform, PP5EnrollTransform)  
405 -  
406 -  
407 -/*!  
408 - * \ingroup distances  
409 - * \brief Compare templates with PP5. PP5 distance is known to be asymmetric  
410 - * \author Josh Klontz \cite jklontz  
411 - * \author E. Taborsky \cite mmtaborsky  
412 - */  
413 -class PP5CompareDistance : public UntrainableDistance  
414 - , public PP5Context  
415 -{  
416 - Q_OBJECT  
417 -  
418 - struct NativeGallery  
419 - {  
420 - FileList files;  
421 - QList<int> subjectIDs;  
422 - ppr_gallery_type gallery;  
423 - };  
424 -  
425 - mutable QMap<QString, NativeGallery> cache;  
426 - mutable QMutex cacheLock;  
427 -  
428 - ~PP5CompareDistance()  
429 - {  
430 - foreach (const NativeGallery &gallery, cache.values())  
431 - ppr_free_gallery(gallery.gallery);  
432 - }  
433 -  
434 - float compare(const cv::Mat &target, const cv::Mat &query) const  
435 - {  
436 - return compare(Template(target), Template(query));  
437 - }  
438 -  
439 - float compare(const Template &target, const Template &query) const  
440 - {  
441 - TemplateList targetList;  
442 - targetList.append(target);  
443 - TemplateList queryList;  
444 - queryList.append(query);  
445 - MatrixOutput *score = MatrixOutput::make(targetList.files(), queryList.files());  
446 - compare(targetList, queryList, score);  
447 - return score->data.at<float>(0);  
448 - }  
449 -  
450 - void compare(const TemplateList &target, const TemplateList &query, Output *output) const  
451 - {  
452 - ppr_gallery_type target_gallery, query_gallery;  
453 - ppr_create_gallery(context, &target_gallery);  
454 - ppr_create_gallery(context, &query_gallery);  
455 - QList<int> target_subject_ids, query_subject_ids;  
456 - enroll(target, &target_gallery, target_subject_ids);  
457 - enroll(query, &query_gallery, query_subject_ids);  
458 - compareNative(target_gallery, target_subject_ids, query_gallery, query_subject_ids, output);  
459 - ppr_free_gallery(target_gallery);  
460 - ppr_free_gallery(query_gallery);  
461 - }  
462 -  
463 - NativeGallery cacheRetain(const File &gallery) const  
464 - {  
465 - QMutexLocker locker(&cacheLock);  
466 - NativeGallery nativeGallery;  
467 - if (cache.contains(gallery.name)) {  
468 - nativeGallery = cache[gallery.name];  
469 - } else {  
470 - ppr_create_gallery(context, &nativeGallery.gallery);  
471 - TemplateList templates = TemplateList::fromGallery(gallery);  
472 - enroll(templates, &nativeGallery.gallery, nativeGallery.subjectIDs);  
473 - nativeGallery.files = templates.files();  
474 - if (gallery.get<bool>("retain"))  
475 - cache.insert(gallery.name, nativeGallery);  
476 - }  
477 - return nativeGallery;  
478 - }  
479 -  
480 - void cacheRelease(const File &gallery, const NativeGallery &nativeGallery) const  
481 - {  
482 - QMutexLocker locker(&cacheLock);  
483 - if (cache.contains(gallery.name)) {  
484 - if (gallery.get<bool>("release")) {  
485 - cache.remove(gallery.name);  
486 - ppr_free_gallery(nativeGallery.gallery);  
487 - }  
488 - } else {  
489 - ppr_free_gallery(nativeGallery.gallery);  
490 - }  
491 - }  
492 -  
493 - bool compare(const File &targetGallery, const File &queryGallery, const File &output) const  
494 - {  
495 - if (!targetGallery.get<bool>("native") || !queryGallery.get<bool>("native"))  
496 - return false;  
497 -  
498 - NativeGallery nativeTarget = cacheRetain(targetGallery);  
499 - NativeGallery nativeQuery = cacheRetain(queryGallery);  
500 -  
501 - QScopedPointer<Output> o(Output::make(output, nativeTarget.files, nativeQuery.files));  
502 - o->setBlock(0, 0);  
503 - compareNative(nativeTarget.gallery, nativeTarget.subjectIDs, nativeQuery.gallery, nativeQuery.subjectIDs, o.data());  
504 -  
505 - cacheRelease(targetGallery, nativeTarget);  
506 - cacheRelease(queryGallery, nativeQuery);  
507 - return true;  
508 - }  
509 -};  
510 -  
511 -BR_REGISTER(Distance, PP5CompareDistance)  
512 -  
513 -/*!  
514 - * \brief DOCUMENT ME  
515 - * \author Unknown \cite unknown  
516 - */  
517 -class PP5GalleryTransform: public UntrainableMetaTransform  
518 - , public PP5Context  
519 -{  
520 - Q_OBJECT  
521 - Q_PROPERTY(QString galleryName READ get_galleryName WRITE set_galleryName RESET reset_galleryName STORED false)  
522 - BR_PROPERTY(QString, galleryName, "")  
523 -  
524 - ppr_gallery_type target;  
525 - QList<int> targetIDs;  
526 - TemplateList gallery;  
527 -  
528 - void project(const Template &src, Template &dst) const  
529 - {  
530 - TemplateList temp, output;  
531 - temp.append(src);  
532 - project(temp, output);  
533 - if (!output.empty())  
534 - dst = output[0];  
535 - }  
536 -  
537 - void project(const TemplateList &src, TemplateList &dst) const  
538 - {  
539 - dst.clear();  
540 - QList<int> queryIDs;  
541 -  
542 - ppr_gallery_type query;  
543 - ppr_create_gallery(context, &query);  
544 - enroll(src,&query, queryIDs);  
545 -  
546 - ppr_similarity_matrix_type simmat;  
547 -  
548 - TRY(ppr_compare_galleries(context, query, target, &simmat))  
549 -  
550 - for (int i=0; i<queryIDs.size(); i++) {  
551 - dst.append(Template());  
552 - dst[i].file = src[i].file;  
553 - dst[i].m() = cv::Mat(1,targetIDs.size(), CV_32FC1);  
554 -  
555 - int query_subject_id = queryIDs[i];  
556 - for (int j=0; j<targetIDs.size(); j++) {  
557 - int target_subject_id = targetIDs[j];  
558 - float score = -std::numeric_limits<float>::max();  
559 - if ((query_subject_id != -1) && (target_subject_id != -1)) {  
560 - TRY(ppr_get_subject_similarity_score(context, simmat, query_subject_id, target_subject_id, &score))  
561 - }  
562 - dst[i].m().at<float>(0,j) = score;  
563 - }  
564 - }  
565 -  
566 - ppr_free_similarity_matrix(simmat);  
567 - ppr_free_gallery(query);  
568 - }  
569 -  
570 - void init()  
571 - {  
572 - if (!galleryName.isEmpty() || !gallery.isEmpty()) {  
573 - // set up the gallery  
574 - ppr_create_gallery(context, &target);  
575 - if (gallery.isEmpty() )  
576 - gallery = TemplateList::fromGallery(galleryName);  
577 - enroll(gallery, &target, targetIDs);  
578 - }  
579 - }  
580 -  
581 - void train(const TemplateList &data)  
582 - {  
583 - gallery = data;  
584 - }  
585 -  
586 - void store(QDataStream &stream) const  
587 - {  
588 - br::Object::store(stream);  
589 - stream << gallery;  
590 - }  
591 -  
592 - void load(QDataStream &stream)  
593 - {  
594 - br::Object::load(stream);  
595 - stream >> gallery;  
596 - init();  
597 - }  
598 -  
599 -};  
600 -  
601 -BR_REGISTER(Transform, PP5GalleryTransform)  
602 -  
603 -#include "classification/pp5.moc"  
openbr/plugins/cmake/pp5.cmake deleted
View file @c023269
1 -set(BR_WITH_PP5 OFF CACHE BOOL "Build with PittPatt 5")  
2 -  
3 -if(${BR_WITH_PP5})  
4 - find_package(PP5 REQUIRED)  
5 - set(BR_THIRDPARTY_LIBS ${BR_THIRDPARTY_LIBS} ${PP5_LIBS})  
6 -  
7 - if(WIN32)  
8 - install(DIRECTORY ${PP5_DIR}/lib/ DESTINATION bin)  
9 - else()  
10 - install(DIRECTORY ${PP5_DIR}/lib/ DESTINATION lib)  
11 - endif()  
12 -  
13 - install(DIRECTORY ${PP5_DIR}/models/ DESTINATION share/openbr/models/pp5)  
14 -else()  
15 - set(BR_EXCLUDED_PLUGINS ${BR_EXCLUDED_PLUGINS} plugins/classification/pp5.cpp)  
16 -endif()