File indexing completed on 2025-03-09 03:54:59

 /* ============================================================
  *
  * This file is a part of digiKam
  *
  * Date        : 02-02-2012
  * Description : Face database interface to train face recognizer.
  *
  * SPDX-FileCopyrightText: 2012-2013 by Marcel Wiesweg <marcel dot wiesweg at gmx dot de>
  * SPDX-FileCopyrightText: 2010-2024 by Gilles Caulier <caulier dot gilles at gmail dot com>
  * SPDX-FileCopyrightText:      2019 by Thanh Trung Dinh <dinhthanhtrung1996 at gmail dot com>
  * SPDX-FileCopyrightText:      2020 by Nghia Duong <minhnghiaduong997 at gmail dot com>
  *
  * SPDX-License-Identifier: GPL-2.0-or-later
  *
  * ============================================================ */
 
 #include "facedb_p.h"
 
 namespace Digikam
 {
 
 int FaceDb::insertFaceVector(const cv::Mat& faceEmbedding,
                              const int label,
                              const QString& context) const
 {
     QVariantList bindingValues;
 
     bindingValues << label;
     bindingValues << context;
     bindingValues << QByteArray::fromRawData((char*)faceEmbedding.ptr<float>(), (sizeof(float) * 128));
 
     DbEngineSqlQuery query = d->db->execQuery(QLatin1String("INSERT INTO FaceMatrices (identity, `context`, embedding) "
                                                             "VALUES (?,?,?);"),
                                               bindingValues);
 
     if (query.lastInsertId().isNull())
     {
         qCWarning(DIGIKAM_FACEDB_LOG) << "fail to insert face embedding, last query"
                                       << query.lastQuery()
                                       << "bound values" << query.boundValues()
                                       << query.lastError();
     }
     else
     {
         qCDebug(DIGIKAM_FACEDB_LOG) << "Commit face mat data "
                                     << query.lastInsertId().toInt()
                                     << " for identity " << label;
     }
 
     return query.lastInsertId().toInt();
 }
 
 KDTree* FaceDb::reconstructTree() const
 {
     KDTree* const tree     = new KDTree(128);
     DbEngineSqlQuery query = d->db->execQuery(QLatin1String("SELECT id, identity, embedding FROM FaceMatrices;"));
 
     while (query.next())
     {
         int nodeId                    = query.value(0).toInt();
         int identity                  = query.value(1).toInt();
         cv::Mat recordedFaceEmbedding = cv::Mat(1, 128, CV_32F, query.value(2).toByteArray().data()).clone();
         KDNode* const newNode         = tree->add(recordedFaceEmbedding, identity);
 
         if (newNode)
         {
             newNode->setNodeId(nodeId);
         }
         else
         {
             qCWarning(DIGIKAM_FACEDB_LOG) << "Error insert node" << nodeId;
         }
     }
 
     return tree;
 }
 
 cv::Ptr<cv::ml::TrainData> FaceDb::trainData() const
 {
     cv::Mat feature, label;
     DbEngineSqlQuery query = d->db->execQuery(QLatin1String("SELECT identity, embedding "
                                                             "FROM FaceMatrices;"));
 
     while (query.next())
     {
         label.push_back(query.value(0).toInt());
         feature.push_back(cv::Mat(1, 128, CV_32F, query.value(1).toByteArray().data()).clone());
     }
 
     return cv::ml::TrainData::create(feature, 0, label);
 }
 
 void FaceDb::clearDNNTraining(const QString& context)
 {
     if (context.isNull())
     {
         d->db->execSql(QLatin1String("DELETE FROM FaceMatrices;"));
     }
     else
     {
         d->db->execSql(QLatin1String("DELETE FROM FaceMatrices WHERE `context`=?;"),
                        context);
     }
 }
 
 void FaceDb::clearDNNTraining(const QList<int>& identities, const QString& context)
 {
     Q_FOREACH (int id, identities)
     {
         if (context.isNull())
         {
             d->db->execSql(QLatin1String("DELETE FROM FaceMatrices WHERE identity=?;"),
                            id);
         }
         else
         {
             d->db->execSql(QLatin1String("DELETE FROM FaceMatrices WHERE identity=? AND `context`=?;"),
                            id, context);
         }
     }
 }
 
 } // namespace Digikam