Nigel
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!18663 SE-net amend utils.cc 

Merge pull request !18663 from chenweitao_295/SE-net_amend
This commit is contained in:
i-robot 2021-06-22 11:48:00 +00:00 committed by Gitee
parent f042392842 06a5aed7d4
commit 11d6b435c2
1 changed files with 8 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
model_zoo/research/cv/SE-Net/ascend310_infer/src/utils.cc
View File

@ -30,17 +30,22 @@ std::vector<std::string> GetAllFiles(std::string_view dirName) {
return {};
}
std::vector<std::string> res;
std::vector<std::string> res_temp;
while ((filename = readdir(dir)) != nullptr) {
std::string dName = std::string(filename->d_name);
if (dName != "." && dName != ".." && filename->d_type == DT_DIR) {
res_temp = GetAllFiles(std::string(dirName) + "/" + filename->d_name);
for (int i = 0; i < static_cast<int>(res_temp.size()); i++) {
res.emplace_back(res_temp[i]);
}
continue;
}
if (dName == "." || dName == ".." || filename->d_type != DT_REG) {
continue;
}
res.emplace_back(std::string(dirName) + "/" + filename->d_name);
}
std::sort(res.begin(), res.end());
for (auto &f : res) {
std::cout << "image file: " << f << std::endl;
}
return res;
}