Added support for the MUCT dataset format for shapes via shapeGallery

Scott Klum
1 parent dc4eaa32
Showing 1 changed file with 73 additions and 0 deletions
openbr/plugins/gallery.cpp
@@ -1588,6 +1588,79 @@ class landmarksGallery : public Gallery
  
 BR_REGISTER(Gallery, landmarksGallery)
  
+/*!
+ * \ingroup galleries
+ * \brief MUCT format for specifying a shape for a file
+ * \author Scott Klum
+ *
+ * http://www.milbo.org/muct/
+ */
+class shapeGallery : public Gallery
+{
+    Q_OBJECT
+
+    QMap<unsigned long, QString> attributes;
+
+    TemplateList readBlock(bool *done)
+    {
+        *done = true;
+        TemplateList templates;
+        QStringList lines = QtUtils::readLines(file);
+        for (int i=0; i<lines.size(); i++) {
+            if (lines[i].at(0) == '#' || lines[i].at(0) == '}') continue;
+
+            // First non-comment line should be "attributes filename"
+            QStringList values = lines[i].split(' ');
+
+            // Assume jpg
+            File file(values.back()+".jpg");
+
+            bool ok;
+            unsigned long attributeKey = values.front().toULong(&ok,16);
+
+            if (attributeKey >= 2147483648) /* Don't use face detection results for now */ {
+                i+=2; continue;
+            } else if (ok && attributes.contains(attributeKey)) {
+                file.set(attributes[attributeKey],true);
+            }
+
+            // Next line should be "{ numLandmarks dimensions"
+            // We don't current support more than two dimensions for landmarks
+            values = lines[++i].split(' ');
+            int numLandmarks = values.at(1).toInt(&ok);
+            if (ok) {
+                QList<QPointF> points; points.reserve(numLandmarks);
+                for (int j=0; j<numLandmarks; j++) {
+                    const QList<float> vals = QtUtils::toFloats(lines[++i].split(' '));
+                    points.append(QPointF(vals[0], vals[1]));
+                }
+                file.setPoints(points);
+            }
+            templates.append(file);
+        }
+        return templates;
+    }
+
+    void write(const Template &t)
+    {
+        (void) t;
+        qFatal("Not implemented.");
+    }
+
+    void init()
+    {
+        // attribute codes
+        attributes[4] = "Glasses";
+        attributes[8] = "Beard";
+        attributes[16] = "Mustache";
+        attributes[256] = "BadImg";
+        attributes[512] = "Cropped";
+        attributes[2048] = "BadEye";
+    }
+};
+
+BR_REGISTER(Gallery, shapeGallery)
+
 #ifdef CVMATIO
  
 using namespace cv;