Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
PaddlePaddle
PaddleSeg
提交
ed41782e
P
PaddleSeg
项目概览
PaddlePaddle
/
PaddleSeg
通知
285
Star
8
Fork
1
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
53
列表
看板
标记
里程碑
合并请求
3
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
P
PaddleSeg
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
53
Issue
53
列表
看板
标记
里程碑
合并请求
3
合并请求
3
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
提交
ed41782e
编写于
12月 16, 2019
作者:
S
sjtubinlong
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
Update C++ infer: search images on windows os
上级
2f36f8cb
变更
1
隐藏空白更改
内联
并排
Showing
1 changed file
with
16 addition
and
9 deletion
+16
-9
deploy/cpp/utils/utils.h
deploy/cpp/utils/utils.h
+16
-9
未找到文件。
deploy/cpp/utils/utils.h
浏览文件 @
ed41782e
...
...
@@ -23,7 +23,8 @@
#include <opencv2/highgui/highgui.hpp>
#ifdef _WIN32
#include <filesystem>
#define GLOG_NO_ABBREVIATED_SEVERITIES
#include <windows.h>
#else
#include <dirent.h>
#include <sys/types.h>
...
...
@@ -67,15 +68,21 @@ namespace utils {
// scan a directory and get all files with input extensions
inline
std
::
vector
<
std
::
string
>
get_directory_images
(
const
std
::
string
&
path
,
const
std
::
string
&
exts
)
{
std
::
string
pattern
(
path
);
pattern
.
append
(
"
\\
*"
);
std
::
vector
<
std
::
string
>
imgs
;
for
(
const
auto
&
item
:
std
::
experimental
::
filesystem
::
directory_iterator
(
path
))
{
auto
suffix
=
item
.
path
().
extension
().
string
();
if
(
exts
.
find
(
suffix
)
!=
std
::
string
::
npos
&&
suffix
.
size
()
>
0
)
{
auto
fullname
=
path_join
(
path
,
item
.
path
().
filename
().
string
());
imgs
.
push_back
(
item
.
path
().
string
());
}
WIN32_FIND_DATA
data
;
HANDLE
hFind
;
if
((
hFind
=
FindFirstFile
(
pattern
.
c_str
(),
&
data
))
!=
INVALID_HANDLE_VALUE
)
{
do
{
auto
fname
=
std
::
string
(
data
.
cFileName
);
auto
pos
=
fname
.
rfind
(
"."
);
auto
ext
=
fname
.
substr
(
pos
+
1
);
if
(
ext
.
size
()
>
1
&&
exts
.
find
(
ext
)
!=
std
::
string
::
npos
)
{
imgs
.
push_back
(
path
+
"
\\
"
+
data
.
cFileName
);
}
}
while
(
FindNextFile
(
hFind
,
&
data
)
!=
0
);
FindClose
(
hFind
);
}
return
imgs
;
}
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录