Minor improvement to example

2016-12-18 20:52:45 -05:00 · 2016-12-18 20:52:45 -05:00 · 093f709a9a
parent ab1575da20
commit 093f709a9a
1 changed files with 2 additions and 1 deletions
--- 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;
 }