Support for FDDB ellipse annotations

Brendan Klare
1 parent 10047515
Showing 1 changed file with 11 additions and 2 deletions
openbr/plugins/gallery.cpp
@@ -856,8 +856,17 @@ class FDDBGallery : public Gallery
             for (int i=0; i<numDetects; i++) {
                 const QStringList detect = lines.takeFirst().split(' ');
                 Template t(fileName);
-                t.file.set("Face", QRectF(detect[0].toFloat(), detect[1].toFloat(), detect[2].toFloat(), detect[3].toFloat()));
-                t.file.set("Confidence", detect[4].toFloat());
+                if (detect.size() == 5) { //rectangle
+                    t.file.set("Face", 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));
+                    t.file.set("Confidence", detect[5].toFloat());
+                } else { //throw an exception?
+                }
                 templates.append(t);
             }
         }