open source pkg v1

pkg/OpenFace/matlab_version/face_detection/detect_faces.m

@@ -0,0 +1,64 @@
+function [ bboxes, shapes ] = detect_faces( image, types )
+%DETECT_FACES Detecting faces in a given image using one of the three detectors
+% INPUT:
+%   image - the image to detect the faces on
+%   type  - cell array of the face detectors to use: 'cascade', 'mtcnn'
+% OUTPUT:
+%   bboxes - a set of bounding boxes describing the detected faces num_faces x
+%   4, the format is [min_x; min_y; max_x; max_y];
+%   shapes - if the face detector detects landmarks as well, output them
+%   n_points x 2 x num_faces
+
+    use_cascade = any(strcmp('cascade', types));
+    use_mtcnn = any(strcmp('mtcnn', types));
+    
+    % Start with fastest one
+    shapes = [];
+    bboxes = [];
+    
+    if(use_cascade)
+        % Check if vision toolbox is available
+        if(~isempty(ver('vision')))
+            face_detector = vision.CascadeObjectDetector();
+            bboxes_det       = step(face_detector, image);
+            if(~isempty(bboxes_det))
+                                                
+                % Convert to the right format
+                bboxes = bboxes_det';
+                
+                % Correct the bounding box to be around face outline
+                % horizontally and from brows to chin vertically
+                % The scalings were learned using the Face Detections on LFPW, Helen, AFW and iBUG datasets
+                % using ground truth and detections from openCV
+
+                % Correct the widths
+                bboxes(3,:) = bboxes(3,:) * 0.8534;
+                bboxes(4,:) = bboxes(4,:) * 0.8972;
+                
+                % Correct the location
+                bboxes(1,:) = bboxes(1,:) + bboxes_det(:,3)'*0.0266;
+                bboxes(2,:) = bboxes(2,:) + bboxes_det(:,4)'*0.1884;
+                
+                bboxes(3,:) = bboxes(1,:) + bboxes(3,:);
+                bboxes(4,:) = bboxes(2,:) + bboxes(4,:);
+            end
+            bboxes = bboxes';
+        else
+            fprintf('Vision toolbox not present in Matlab, it is necessary for Cascade face detection\n');
+        end
+    end
+    
+    if(use_mtcnn && isempty(bboxes))
+        od = cd('../face_detection/mtcnn/');
+
+        if(~exist('PNet_mlab.mat', 'file'))
+            fprintf('WARNING: Could not find MTCNN model files, they are needed for face detection. Use Matlab cascade detector if they are not available\n');
+            return;
+        end
+        
+        [bboxes, shapes] = detect_face_mtcnn(image);
+        cd(od);        
+    end
+    
+end
+