Server: add option to disable face recognition

Author: fuzun

server/source/Client/Client.cpp

@@ -173,6 +173,14 @@ void Client::processTextMessage(const QString& string)
             sendCommand(Command::SETTING_DETERMINISTICOBJECTDETECTION, settings.deterministicObjectDetection.load());
             break;
         }
+
+        case (unsigned int)Command::SETTING_FACERECOGNITIONENABLED:
+        {
+            settings.faceRecognitionEnabled = context.toBool();
+
+            sendCommand(Command::SETTING_FACERECOGNITIONENABLED, settings.faceRecognitionEnabled.load());
+            break;
+        }
         // implement later ...
     }
 }

server/source/Client/Client.h

@@ -28,6 +28,7 @@ struct Settings
     std::atomic<bool> objectDetectionEnabled = true;
     std::atomic<size_t> labelCount = 5;
     std::atomic<bool> deterministicObjectDetection = false;
+    std::atomic<bool> faceRecognitionEnabled = true;
 };
 
 class Client : public QObject
@@ -43,7 +44,8 @@ class Client : public QObject
         MESSAGE_TAG_OBJECT = 4,
         SETTING_OBJDETECTIONENABLED = 5,
         SETTING_LABELCOUNT = 6,
-        SETTING_DETERMINISTICOBJECTDETECTION = 7
+        SETTING_DETERMINISTICOBJECTDETECTION = 7,
+        SETTING_FACERECOGNITIONENABLED = 8
     };
 
     Settings settings;

server/source/DLIBWorker/DLIBWorker.cpp

@@ -297,31 +297,35 @@ void DLIBWorker::process(const QByteArray& buffer)
             emit doneObject(result);
         }
 
-        auto faces = findFaces(img);
-
-        rectangle rect;
-        for (const auto& face : referenceFaces)
+        if (m_settings->faceRecognitionEnabled)
         {
-            faces.push_back(make_tuple(get<0>(face), rect, get<2>(face)));
-        }
+            auto faces = findFaces(img);
 
-        auto graph = createGraph(faces, m_threshold);
+            rectangle rect;
+            for (const auto& face : referenceFaces)
+            {
+                faces.push_back(make_tuple(get<0>(face), rect, get<2>(face)));
+            }
 
-        auto clusters = cluster(graph, faces);
+            auto graph = createGraph(faces, m_threshold);
 
-        QVector<QPair<QRect, QString>> linearFaces;
+            auto clusters = cluster(graph, faces);
 
-        for (const auto& it : clusters)
-        {
-            QString str = it.first;
-            for (const auto& it2 : it.second)
+            QVector<QPair<QRect, QString>> linearFaces;
+
+            for (const auto& it : clusters)
             {
-                linearFaces.push_back(qMakePair(QRect(it2.left(), it2.top(), it2.width(), it2.height()), str));
+                QString str = it.first;
+                for (const auto& it2 : it.second)
+                {
+                    linearFaces.push_back(qMakePair(QRect(it2.left(), it2.top(), it2.width(), it2.height()), str));
+                }
             }
+
+            emit doneFace(linearFaces);
         }
 
         m_busy = false;
-        emit doneFace(linearFaces);
     }
     catch(const std::exception& e)
     {