Merge pull request #180 from lifengcai/master

fix list mnn models bug on Windows platform
alibaba · Jul 25, 2019 · 28245fc · 28245fc
2 parents 521a664 + 933f8bc
commit 28245fc
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/benchmark/benchmark.cpp b/benchmark/benchmark.cpp
@@ -54,7 +54,8 @@ std::vector<Model> findModelFiles(const char* dir) {
 #if defined(_MSC_VER)
     WIN32_FIND_DATA ffd;
     HANDLE hFind = INVALID_HANDLE_VALUE;
-    hFind = FindFirstFile(dir, &ffd);
+    std::string mnn_model_pattern = std::string(dir) + "\\*.mnn"; 
+    hFind = FindFirstFile(mnn_model_pattern.c_str(), &ffd);
     if (INVALID_HANDLE_VALUE == hFind) {
         std::cout << "open " << dir << " failed: " << strerror(errno) << std::endl;
         return models;