Minor improvement to example

093f709a · Davis King · ab1575da · 093f709a
Commit 093f709a authored Dec 19, 2016 by Davis King
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

dnn_metric_learning_on_images_ex.cpp examples/dnn_metric_learning_on_images_ex.cpp +2 -1

No files found.
--- a/examples/dnn_metric_learning_on_images_ex.cpp
+++ b/examples/dnn_metric_learning_on_images_ex.cpp
@@ -67,7 +67,8 @@ std::vector<std::vector<string>> load_objects_list (
        for (auto img : subdir.get_files())
            imgs.push_back(img);

-        objects.push_back(imgs);
+        if (imgs.size() != 0)
+            objects.push_back(imgs);
    }
    return objects;
 }