openfacepreprocessor.h

Go to the documentation of this file.

/* ============================================================
 *
 * This file is a part of digiKam project
 * https://www.digikam.org
 *
 * Date        : 2019-07-09
 * Description : Preprocessor for openface nn model
 *
 * Copyright (C) 2019      by Thanh Trung Dinh <dinhthanhtrung1996 at gmail dot com>
 * Copyright (C) 2019-2022 by Gilles Caulier <caulier dot gilles at gmail dot com>
 *
 * This program is free software; you can redistribute it
 * and/or modify it under the terms of the GNU General
 * Public License as published by the Free Software Foundation;
 * either version 2, or (at your option)
 * any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * ============================================================ */
 
#ifndef DIGIKAM_OPENFACE_PREPROCESSOR_H
#define DIGIKAM_OPENFACE_PREPROCESSOR_H
 
// C++ includes
 
#include <array>
 
// Qt includes
 
#include <QMutex>
 
// Local includes
 
#include "digikam_opencv.h"
#include "shapepredictor.h"
 
namespace Digikam
{
 
class OpenfacePreprocessor
{
 
public:
 
    explicit OpenfacePreprocessor();
    ~OpenfacePreprocessor();
 
    bool loadModels();
 
    cv::Mat process(const cv::Mat& image);
 
private:
 
    cv::Size               outImageSize;
 
    cv::Mat                faceTemplate;
    std::array<int, 3>     outerEyesNosePositions;
 
    RedEye::ShapePredictor sp;
 
    QMutex                 mutex;
 
private:
 
    // Disable
    OpenfacePreprocessor(const OpenfacePreprocessor&)            = delete;
    OpenfacePreprocessor& operator=(const OpenfacePreprocessor&) = delete;
};
 
} // namespace Digikam
 
#endif // DIGIKAM_OPENFACE_PREPROCESSOR_H