Peter M. Groen / openbr

Commit 9b8db88d974fff3c74b9d155f9b51bff595c8bfd

Authored by Josh Klontz
1 parent 780b1296

removed dlib

Inline Side-by-side
Showing 3 changed files with 0 additions and 226 deletions
openbr/plugins/classification/dlib.cpp deleted
View file @780b129
1   -#include <opencv2/imgproc/imgproc.hpp>
2   -#include <dlib/image_processing/frontal_face_detector.h>
3   -#include <dlib/image_processing.h>
4   -#include <dlib/opencv.h>
5   -
6   -#include "openbr/plugins/openbr_internal.h"
7   -
8   -#include <QTemporaryFile>
9   -
10   -using namespace std;
11   -using namespace dlib;
12   -
13   -namespace br
14   -{
15   -
16   -/*!
17   - * \ingroup transforms
18   - * \brief Wrapper to dlib's landmarker.
19   - * \author Scott Klum \cite sklum
20   - */
21   -class DLibShapeResourceMaker : public ResourceMaker<shape_predictor>
22   -{
23   -
24   -private:
25   - shape_predictor *make() const
26   - {
27   - shape_predictor *sp = new shape_predictor();
28   - dlib::deserialize(qPrintable(Globals->sdkPath + "/share/openbr/models/dlib/shape_predictor_68_face_landmarks.dat")) >> *sp;
29   - return sp;
30   - }
31   -};
32   -
33   -class DLandmarkerTransform : public UntrainableTransform
34   -{
35   - Q_OBJECT
36   -
37   -private:
38   - Resource<shape_predictor> shapeResource;
39   -
40   - void init()
41   - {
42   - shapeResource.setResourceMaker(new DLibShapeResourceMaker());
43   - shapeResource.release(shapeResource.acquire()); // Pre-load one instance of the model
44   - }
45   -
46   - QPointF averagePoints(const QList<QPointF> &points, int rangeBegin, int rangeEnd) const
47   - {
48   - QPointF point;
49   - for (int i=rangeBegin; i<rangeEnd; i++)
50   - point += points[i];
51   - point /= (rangeEnd-rangeBegin);
52   - return point;
53   - }
54   -
55   - void setFacePoints(Template &dst) const
56   - {
57   - const QList<QPointF> points = dst.file.points();
58   - dst.file.set("RightEye", averagePoints(points, 36, 42));
59   - dst.file.set("LeftEye" , averagePoints(points, 42, 48));
60   - dst.file.set("Chin", points[8]);
61   - }
62   -
63   - void project(const Template &src, Template &dst) const
64   - {
65   - dst = src;
66   -
67   - shape_predictor *const sp = shapeResource.acquire();
68   -
69   - cv::Mat cvImage = src.m();
70   - if (cvImage.channels() == 3)
71   - cv::cvtColor(cvImage, cvImage, CV_BGR2GRAY);
72   -
73   - cv_image<unsigned char> cimg(cvImage);
74   - array2d<unsigned char> image;
75   - assign_image(image,cimg);
76   -
77   - rectangle r;
78   - if (src.file.rects().isEmpty()) { // If the image has no rects assume the whole image is a face
79   - r = rectangle(0, 0, cvImage.cols, cvImage.rows);
80   - } else { // Crop the image on the first rect
81   - const QRectF rect = src.file.rects().first();
82   - r = rectangle(rect.left(), rect.top(), rect.right(), rect.bottom());
83   - }
84   -
85   - full_object_detection shape = (*sp)(image, r);
86   - QList<QPointF> points;
87   - for (size_t i=0; i<shape.num_parts(); i++)
88   - points.append(QPointF(shape.part(i)(0), shape.part(i)(1)));
89   - dst.file.setPoints(points);
90   - setFacePoints(dst);
91   -
92   - shapeResource.release(sp);
93   - }
94   -};
95   -
96   -BR_REGISTER(Transform, DLandmarkerTransform)
97   -
98   -/*!
99   - * \ingroup transforms
100   - * \brief Wrapper to dlib's trainable object detector.
101   - * \author Scott Klum \cite sklum
102   - */
103   -class DObjectDetectorTransform : public Transform
104   -{
105   - Q_OBJECT
106   -
107   - Q_PROPERTY(int winSize READ get_winSize WRITE set_winSize RESET reset_winSize STORED true)
108   - Q_PROPERTY(float C READ get_C WRITE set_C RESET reset_C STORED true)
109   - Q_PROPERTY(float epsilon READ get_epsilon WRITE set_epsilon RESET reset_epsilon STORED true)
110   - BR_PROPERTY(int, winSize, 80)
111   - BR_PROPERTY(float, C, 1)
112   - BR_PROPERTY(float, epsilon, .01)
113   -
114   -private:
115   - typedef scan_fhog_pyramid<pyramid_down<6> > image_scanner_type;
116   - mutable object_detector<image_scanner_type> detector;
117   - mutable QMutex mutex;
118   -
119   - void train(const TemplateList &data)
120   - {
121   - dlib::array<array2d<unsigned char> > samples;
122   - std::vector<std::vector<rectangle> > boxes;
123   -
124   - foreach (const Template &t, data) {
125   - if (!t.file.rects().isEmpty()) {
126   - cv_image<unsigned char> cimg(t.m());
127   -
128   - array2d<unsigned char> image;
129   - assign_image(image,cimg);
130   -
131   - samples.push_back(image);
132   -
133   - std::vector<rectangle> b;
134   - foreach (const QRectF &r, t.file.rects())
135   - b.push_back(rectangle(r.left(),r.top(),r.right(),r.bottom()));
136   -
137   - boxes.push_back(b);
138   - }
139   - }
140   -
141   - if (samples.size() == 0)
142   - qFatal("Training data has no bounding boxes.");
143   -
144   - image_scanner_type scanner;
145   -
146   - scanner.set_detection_window_size(winSize, winSize);
147   - structural_object_detection_trainer<image_scanner_type> trainer(scanner);
148   - trainer.set_num_threads(max(1,QThread::idealThreadCount()));
149   - trainer.set_c(C);
150   - trainer.set_epsilon(epsilon);
151   -
152   - if (Globals->verbose)
153   - trainer.be_verbose();
154   -
155   - detector = trainer.train(samples, boxes);
156   - }
157   -
158   - void project(const Template &src, Template &dst) const
159   - {
160   - dst = src;
161   - cv_image<unsigned char> cimg(src.m());
162   - array2d<unsigned char> image;
163   - assign_image(image,cimg);
164   -
165   - QMutexLocker locker(&mutex);
166   - std::vector<rectangle> dets = detector(image);
167   - locker.unlock();
168   -
169   - for (size_t i=0; i<dets.size(); i++)
170   - dst.file.appendRect(QRectF(QPointF(dets[i].left(),dets[i].top()),QPointF(dets[i].right(),dets[i].bottom())));
171   - }
172   -
173   - void store(QDataStream &stream) const
174   - {
175   - // Create local file
176   - QTemporaryFile tempFile;
177   - tempFile.open();
178   - tempFile.close();
179   -
180   - dlib::serialize(qPrintable(tempFile.fileName())) << detector;
181   -
182   - // Copy local file contents to stream
183   - tempFile.open();
184   - QByteArray data = tempFile.readAll();
185   - tempFile.close();
186   - stream << data;
187   - }
188   -
189   - void load(QDataStream &stream)
190   - {
191   - // Copy local file contents from stream
192   - QByteArray data;
193   - stream >> data;
194   -
195   - // Create local file
196   - QTemporaryFile tempFile(QDir::tempPath()+"/model");
197   - tempFile.open();
198   - tempFile.write(data);
199   - tempFile.close();
200   -
201   - // Load MLP from local file
202   - dlib::deserialize(qPrintable(tempFile.fileName())) >> detector;
203   - }
204   -};
205   -
206   -BR_REGISTER(Transform, DObjectDetectorTransform)
207   -
208   -} // namespace br
209   -
210   -#include "dlib.moc"
openbr/plugins/classification/dlib_source.cpp deleted
View file @780b129
1   -#define DLIB_NO_GUI_SUPPORT
2   -#include <dlib/all/source.cpp>
openbr/plugins/cmake/dlib.cmake deleted
View file @780b129
1   -set(BR_WITH_DLIB OFF CACHE BOOL "Build with DLib")
2   -
3   -if(${BR_WITH_DLIB})
4   - ExternalProject_Add(dlib
5   - URL http://downloads.sourceforge.net/project/dclib/dlib/v18.16/dlib-18.16.tar.bz2
6   - URL_MD5 e9e5449bc25370afce2d254327afac99
7   - SOURCE_DIR "${PROJECT_SOURCE_DIR}/3rdparty/dlib-18.16"
8   - CONFIGURE_COMMAND ""
9   - BUILD_COMMAND ""
10   - INSTALL_COMMAND "")
11   - include_directories("${PROJECT_SOURCE_DIR}/3rdparty/dlib-18.16")
12   -else()
13   - set(BR_EXCLUDED_PLUGINS ${BR_EXCLUDED_PLUGINS} plugins/classification/dlib.cpp plugins/classification/dlib_source.cpp)
14   -endif()