face_alignment.cpp

/*
 *
 * This file is part of the open-source SeetaFace engine, which includes three modules:
 * SeetaFace Detection, SeetaFace Alignment, and SeetaFace Identification.
 *
 * This file is part of the SeetaFace Alignment module, containing codes implementing the
 * facial landmarks location method described in the following paper:
 *
 *
 *   Coarse-to-Fine Auto-Encoder Networks (CFAN) for Real-Time Face Alignment, 
 *   Jie Zhang, Shiguang Shan, Meina Kan, Xilin Chen. In Proceeding of the
 *   European Conference on Computer Vision (ECCV), 2014
 *
 *
 * Copyright (C) 2016, Visual Information Processing and Learning (VIPL) group,
 * Institute of Computing Technology, Chinese Academy of Sciences, Beijing, China.
 *
 * The codes are mainly developed by Jie Zhang (a Ph.D supervised by Prof. Shiguang Shan)
 *
 * As an open-source face recognition engine: you can redistribute SeetaFace source codes
 * and/or modify it under the terms of the BSD 2-Clause License.
 *
 * You should have received a copy of the BSD 2-Clause License along with the software.
 * If not, see < https://opensource.org/licenses/BSD-2-Clause>.
 *
 * Contact Info: you can send an email to SeetaFace@vipl.ict.ac.cn for any problems.
 *
 * Note: the above information must be kept whenever or wherever the codes are used.
 *
 */

#include "face_alignment.h"

#include <string>
#include <math.h>
#include "cfan.h"

namespace seeta {
  /** A constructor with an optional argument specifying path of the model file.
   *  If called with no argument, the model file is assumed to be stored in the
   *  the working directory as "seeta_fa_v1.1.bin".
   *
   *  @param model_path Path of the model file, either absolute or relative to
   *  the working directory.
   */
  FaceAlignment::FaceAlignment(const char * model_path){
    facial_detector = new CCFAN();
    if (model_path == NULL)
      model_path = "seeta_fa_v1.1.bin";
    facial_detector->InitModel(model_path);
  }

  /** Detect five facial landmarks, i.e., two eye centers, nose tip and two mouth corners.
   *  @param gray_im A grayscale image
   *  @param face_info The face bounding box
   *  @param[out] points The locations of detected facial points
   */
  bool FaceAlignment::PointDetectLandmarks(ImageData gray_im, FaceInfo face_info, FacialLandmark *points)
  {
    if (gray_im.num_channels != 1) {
      return false;
    }
    int pts_num = 5;
    float *facial_loc = new float[pts_num * 2];
    facial_detector->FacialPointLocate(gray_im.data, gray_im.width, gray_im.height, face_info, facial_loc);

    for (int i = 0; i < pts_num; i++) {
      points[i].x = facial_loc[i * 2];
      points[i].y = facial_loc[i * 2 + 1];
    }

    delete[]facial_loc;
    return true;
  }

  /** A Destructor which should never be called explicitly.
   *  Release all dynamically allocated resources.
   */
  FaceAlignment::~FaceAlignment() {
    if (facial_detector != NULL) {
      delete facial_detector;
      facial_detector = NULL;
    }
  }
}  // namespace seeta