Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
urhero
Paddle
提交
250206d1
P
Paddle
项目概览
urhero
/
Paddle
与 Fork 源项目一致
Fork自
PaddlePaddle / Paddle
通知
1
Star
0
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
P
Paddle
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
体验新版 GitCode,发现更多精彩内容 >>
提交
250206d1
编写于
1月 23, 2018
作者:
L
Liu Yiqun
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
Change the example of inference to a unittest.
上级
c6482444
变更
3
显示空白变更内容
内联
并排
Showing
3 changed file
with
31 addition
and
33 deletion
+31
-33
paddle/inference/CMakeLists.txt
paddle/inference/CMakeLists.txt
+2
-15
paddle/inference/tests/book/CMakeLists.txt
paddle/inference/tests/book/CMakeLists.txt
+13
-0
paddle/inference/tests/book/test_inference_recognize_digits_mlp.cc
...ference/tests/book/test_inference_recognize_digits_mlp.cc
+16
-18
未找到文件。
paddle/inference/CMakeLists.txt
浏览文件 @
250206d1
...
...
@@ -24,19 +24,6 @@ if(NOT WITH_C_API AND WITH_FLUID)
install
(
TARGETS paddle_fluid_shared DESTINATION lib
)
endif
()
add_executable
(
example example.cc
)
if
(
APPLE
)
set
(
OPTIONAL_LINK_FLAGS
)
if
(
"
${
CMAKE_CXX_COMPILER_ID
}
"
STREQUAL
"Clang"
OR
"
${
CMAKE_CXX_COMPILER_ID
}
"
STREQUAL
"AppleClang"
)
set
(
OPTIONAL_LINK_FLAGS
"-undefined dynamic_lookup"
)
endif
()
target_link_libraries
(
example
-Wl,-force_load paddle_fluid
${
OPTIONAL_LINK_FLAGS
}
${
PTOOLS_LIB
}
)
else
()
target_link_libraries
(
example
-Wl,--start-group -Wl,--whole-archive paddle_fluid
-Wl,--no-whole-archive -Wl,--end-group
${
PTOOLS_LIB
}
)
if
(
WITH_TESTING
)
add_subdirectory
(
tests/book
)
endif
()
paddle/inference/tests/book/CMakeLists.txt
0 → 100644
浏览文件 @
250206d1
add_executable
(
test_inference_recognize_digits_mlp test_inference_recognize_digits_mlp.cc
)
target_circle_link_libraries
(
test_inference_recognize_digits_mlp
ARCHIVE_START
paddle_fluid
ARCHIVE_END
gtest
gflags
)
add_test
(
NAME test_inference_recognize_digits_mlp
COMMAND test_inference_recognize_digits_mlp
--dirname=
${
PADDLE_SOURCE_DIR
}
/python/paddle/v2/fluid/tests/book/recognize_digits_mlp.inference.model
WORKING_DIRECTORY
${
CMAKE_CURRENT_SOURCE_DIR
}
)
paddle/inference/
example
.cc
→
paddle/inference/
tests/book/test_inference_recognize_digits_mlp
.cc
浏览文件 @
250206d1
...
...
@@ -12,20 +12,17 @@ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License. */
#include <gtest/gtest.h>
#include <time.h>
#include <
io
stream>
#include <
s
stream>
#include "gflags/gflags.h"
#include "paddle/inference/inference.h"
DEFINE_string
(
dirname
,
""
,
"Directory of the inference model."
);
int
main
(
int
argc
,
char
**
argv
)
{
google
::
ParseCommandLineFlags
(
&
argc
,
&
argv
,
true
);
TEST
(
inference
,
recognize_digits_mlp
)
{
if
(
FLAGS_dirname
.
empty
())
{
// Example:
// ./example --dirname=recognize_digits_mlp.inference.model
std
::
cout
<<
"Usage: ./example --dirname=path/to/your/model"
<<
std
::
endl
;
exit
(
1
);
LOG
(
FATAL
)
<<
"Usage: ./example --dirname=path/to/your/model"
;
}
std
::
cout
<<
"FLAGS_dirname: "
<<
FLAGS_dirname
<<
std
::
endl
;
...
...
@@ -48,20 +45,21 @@ int main(int argc, char** argv) {
engine
->
Execute
(
feeds
,
fetchs
);
for
(
size_t
i
=
0
;
i
<
fetchs
.
size
();
++
i
)
{
auto
dims_i
=
fetchs
[
i
].
dims
();
std
::
cout
<<
"dims_i:"
;
for
(
int
j
=
0
;
j
<
dims_i
.
size
();
++
j
)
{
std
::
cout
<<
" "
<<
dims_i
[
j
];
}
std
::
cout
<<
std
::
endl
;
std
::
cout
<<
"result:"
;
LOG
(
INFO
)
<<
fetchs
[
i
].
dims
();
std
::
stringstream
ss
;
ss
<<
"result:"
;
float
*
output_ptr
=
fetchs
[
i
].
data
<
float
>
();
for
(
int
j
=
0
;
j
<
paddle
::
framework
::
product
(
dims_i
);
++
j
)
{
s
td
::
cout
<<
" "
<<
output_ptr
[
j
];
for
(
int
j
=
0
;
j
<
fetchs
[
i
].
numel
(
);
++
j
)
{
s
s
<<
" "
<<
output_ptr
[
j
];
}
std
::
cout
<<
std
::
endl
;
LOG
(
INFO
)
<<
ss
.
str
()
;
}
delete
engine
;
return
0
;
}
int
main
(
int
argc
,
char
**
argv
)
{
google
::
ParseCommandLineFlags
(
&
argc
,
&
argv
,
false
);
testing
::
InitGoogleTest
(
&
argc
,
argv
);
return
RUN_ALL_TESTS
();
}
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录