FDDB import to list of rects for slidingWindow compatability

Brendan Klare
1 parent f83ba8b7
Showing 1 changed file with 5 additions and 2 deletions
openbr/plugins/gallery.cpp
@@ -857,18 +857,21 @@ class FDDBGallery : public Gallery
             for (int i=0; i<numDetects; i++) {
                 const QStringList detect = lines.takeFirst().split(' ');
                 Template t(fileName);
+                QList<QVariant> faceList; //to be consistent with slidingWindow
                 if (detect.size() == 5) { //rectangle
-                    t.file.set("Face", QRectF(detect[0].toFloat(), detect[1].toFloat(), detect[2].toFloat(), detect[3].toFloat()));
+                    faceList.append(QRectF(detect[0].toFloat(), detect[1].toFloat(), detect[2].toFloat(), detect[3].toFloat()));
                     t.file.set("Confidence", detect[4].toFloat());
                 } else if (detect.size() == 6) { //ellipse
                     float x = detect[3].toFloat(),  
                           y = detect[4].toFloat(),
                           radius = detect[1].toFloat();
-                    t.file.set("Face", QRectF(x - radius,y - radius,radius * 2.0, radius * 2.0));
+                    faceList.append(QRectF(x - radius,y - radius,radius * 2.0, radius * 2.0));
                     t.file.set("Confidence", detect[5].toFloat());
                 } else {
                     qFatal("Unknown FDDB annotation format.");
                 }
+                t.file.set("Face", faceList);
+                t.file.set("Label",QString("face"));
                 templates.append(t);
             }
         }