Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
飞不过沧海
Paddle
提交
f6a877bc
P
Paddle
项目概览
飞不过沧海
/
Paddle
与 Fork 源项目一致
Fork自
PaddlePaddle / Paddle
通知
1
Star
1
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,发现更多精彩内容 >>
提交
f6a877bc
编写于
12月 06, 2018
作者:
F
flame
提交者:
Yan Chunwei
12月 06, 2018
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
add tool to visualize inference model (#14621)
上级
b52f5d28
变更
3
隐藏空白更改
内联
并排
Showing
3 changed file
with
139 addition
and
0 deletion
+139
-0
paddle/fluid/inference/utils/CMakeLists.txt
paddle/fluid/inference/utils/CMakeLists.txt
+5
-0
paddle/fluid/inference/utils/visualizer.cc
paddle/fluid/inference/utils/visualizer.cc
+92
-0
paddle/fluid/inference/utils/visualizer.h
paddle/fluid/inference/utils/visualizer.h
+42
-0
未找到文件。
paddle/fluid/inference/utils/CMakeLists.txt
浏览文件 @
f6a877bc
cc_library
(
benchmark SRCS benchmark.cc DEPS enforce
)
cc_test
(
test_benchmark SRCS benchmark_tester.cc DEPS benchmark
)
cc_binary
(
visualizer SRCS visualizer.cc DEPS analysis
paddle_pass_builder ir_pass_manager pass graph_viz_pass analysis_passes
)
if
(
WIN32
)
target_link_libraries
(
visualizer shlwapi
)
endif
(
WIN32
)
paddle/fluid/inference/utils/visualizer.cc
0 → 100644
浏览文件 @
f6a877bc
// Copyright (c) 2018 PaddlePaddle Authors. All Rights Reserved.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// 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 "paddle/fluid/inference/utils/visualizer.h"
#include <gflags/gflags.h>
#include <glog/logging.h>
#include <fstream>
#include <memory>
#include "paddle/fluid/framework/ir/graph_viz_pass.h"
#include "paddle/fluid/inference/analysis/analyzer.h"
#include "paddle/fluid/inference/analysis/passes/ir_analysis_pass.h"
#include "paddle/fluid/platform/init.h"
DEFINE_string
(
model_dir
,
""
,
"model directory"
);
DEFINE_string
(
model_program_path
,
""
,
"model program path"
);
DEFINE_string
(
model_params_path
,
""
,
"model params path"
);
USE_PASS
(
graph_viz_pass
);
USE_PASS
(
graph_to_program_pass
);
using
paddle
::
inference
::
analysis
::
Argument
;
namespace
paddle
{
namespace
inference
{
namespace
utils
{
void
Visualizer
::
SetArgument
(
Argument
*
argument
)
{
argument_
=
argument
;
}
bool
Visualizer
::
Run
()
{
paddle
::
framework
::
InitDevices
(
false
);
paddle
::
inference
::
analysis
::
Analyzer
().
Run
(
argument_
);
return
true
;
}
}
// namespace utils
}
// namespace inference
}
// namespace paddle
// Generate a dot file describing the structure of graph.
// To use this tool, run command: ./visualizer [options...]
// Options:
// --model_dir: the directory of model
// --model_program_path: the path of program
// --model_params_path: the path of params
int
main
(
int
argc
,
char
*
argv
[])
{
gflags
::
ParseCommandLineFlags
(
&
argc
,
&
argv
,
true
);
google
::
InitGoogleLogging
(
argv
[
0
]);
paddle
::
inference
::
analysis
::
Argument
argument
;
argument
.
SetUseGPU
(
false
);
argument
.
SetUseTensorRT
(
false
);
if
(
FLAGS_model_dir
.
empty
())
{
if
(
FLAGS_model_program_path
.
empty
()
||
FLAGS_model_params_path
.
empty
())
{
LOG
(
ERROR
)
<<
"Please set model_dir"
" or model_program_path and model_params_path"
;
return
-
1
;
}
else
{
argument
.
SetModelProgramPath
(
FLAGS_model_program_path
);
argument
.
SetModelParamsPath
(
FLAGS_model_params_path
);
}
}
else
{
argument
.
SetModelDir
(
FLAGS_model_dir
);
}
// Only 1 pass, default filename is 0_ir_origin.dot
// For more details, looking for paddle::inference::analysis::IRPassManager
argument
.
SetIrAnalysisPasses
({
"graph_viz_pass"
});
std
::
unique_ptr
<
paddle
::
framework
::
Scope
>
scope
{
new
paddle
::
framework
::
Scope
()};
argument
.
SetScopeNotOwned
(
const_cast
<
paddle
::
framework
::
Scope
*>
(
scope
.
get
()));
paddle
::
inference
::
utils
::
Visualizer
visualizer
;
visualizer
.
SetArgument
(
&
argument
);
visualizer
.
Run
();
return
0
;
}
paddle/fluid/inference/utils/visualizer.h
0 → 100644
浏览文件 @
f6a877bc
// Copyright (c) 2018 PaddlePaddle Authors. All Rights Reserved.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// 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.
#pragma once
#include <string>
#include "paddle/fluid/inference/analysis/argument.h"
namespace
paddle
{
namespace
inference
{
namespace
utils
{
using
paddle
::
inference
::
analysis
::
Argument
;
class
Visualizer
final
{
public:
Visualizer
()
=
default
;
~
Visualizer
()
=
default
;
Visualizer
(
const
Visualizer
&
)
=
delete
;
Visualizer
&
operator
=
(
const
Visualizer
&
)
=
delete
;
void
SetArgument
(
Argument
*
);
bool
Run
();
private:
Argument
*
argument_
;
};
}
// namespace utils
}
// namespace inference
}
// namespace paddle
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录