More efficient hash-based search in getNumberOfImages

Jordan Cheney
1 parent 19a31ced
Showing 1 changed file with 4 additions and 5 deletions
openbr/core/eval.cpp
@@ -747,12 +747,11 @@ static QMap&lt;QString, Detections&gt; getDetections(const File &amp;predictedGallery, con
 static int getNumberOfImages(const File &truthGallery)
 {
-    const FileList files = TemplateList::fromGallery(truthGallery).files();
+    const FileList files = FileList::fromGallery(truthGallery);
-    QStringList names;
-    foreach(const File file, files) 
-        if (!names.contains(file.fileName()))
-            names.append(file.fileName());
+    QSet<QString> names;
+    foreach(const File &file, files)
+        names.insert(file.fileName());
     return names.size();
 }