Add very crude plugin to detect faces in all images

2026-07-14 11:18:22 +08:00 · 2011-06-07 18:35:03 +02:00
parent 6d48564295
commit 628c9da5b3
1 changed files with 36 additions and 0 deletions
@@ -0,0 +1,36 @@
+from sloth.annotations.model import ImageRole
+from PyQt4.QtGui import *
+from PyQt4.QtCore import *
+from okapy import BinaryPatternFaceDetector
+
+class FaceDetectorPlugin(QObject):
+    def __init__(self, wnd):
+        QObject.__init__(self, wnd)
+        self.wnd_ = wnd
+        self.sc_  = QAction("Detect faces", wnd)
+        self.sc_.triggered.connect(self.doit)
+
+    def doit(self):
+        print "Loading detector..."
+        det = BinaryPatternFaceDetector("/cvhci/data/mctcascades/new-detectors/face_frontal_new.xml")
+        model = self.wnd_.model_
+        n_images = model.rowCount()
+        for i in range(n_images):
+            index = model.index(i, 0)
+            item = model.itemFromIndex(index)
+            img = item.data(index, ImageRole)
+            faces = det.detectFaces(img)
+            for face in faces:
+                ann = {
+                    'type':       'rect',
+                    'class':      'face',
+                    'x':          face.box.x,
+                    'y':          face.box.y,
+                    'width':      face.box.width,
+                    'height':     face.box.height,
+                    'confidence': face.conf,
+                }
+                model.addAnnotation(index, ann)
+
+    def action(self):
+        return self.sc_