Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
Greenplum
Opencv
提交
12fdaf89
O
Opencv
项目概览
Greenplum
/
Opencv
10 个月 前同步成功
通知
7
Star
0
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
DevOps
流水线
流水线任务
计划
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
O
Opencv
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
DevOps
DevOps
流水线
流水线任务
计划
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
流水线任务
提交
Issue看板
前往新版Gitcode,体验更适合开发者的 AI 搜索 >>
提交
12fdaf89
编写于
7月 18, 2019
作者:
L
Lubov Batanina
提交者:
Alexander Alekhin
7月 18, 2019
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
Merge pull request #15057 from l-bat:fix_vizualizer
* Fix dumpToFile * Add test * Fix test
上级
c12e26ff
变更
3
隐藏空白更改
内联
并排
Showing
3 changed file
with
28 addition
and
1 deletion
+28
-1
modules/dnn/include/opencv2/dnn/dnn.hpp
modules/dnn/include/opencv2/dnn/dnn.hpp
+1
-1
modules/dnn/src/dnn.cpp
modules/dnn/src/dnn.cpp
+7
-0
modules/dnn/test/test_misc.cpp
modules/dnn/test/test_misc.cpp
+20
-0
未找到文件。
modules/dnn/include/opencv2/dnn/dnn.hpp
浏览文件 @
12fdaf89
...
...
@@ -387,7 +387,7 @@ CV__DNN_EXPERIMENTAL_NS_BEGIN
/** @brief Dump net to String
* @returns String with structure, hyperparameters, backend, target and fusion
* To see correct backend, target and fusion run after forward().
*
Call method after setInput().
To see correct backend, target and fusion run after forward().
*/
CV_WRAP
String
dump
();
/** @brief Dump net structure, hyperparameters, backend, target and fusion to dot file
...
...
modules/dnn/src/dnn.cpp
浏览文件 @
12fdaf89
...
...
@@ -2903,6 +2903,13 @@ String parseLayerParams(const String& name, const LayerParams& lp) {
String
Net
::
dump
()
{
CV_Assert
(
!
empty
());
if
(
impl
->
netInputLayer
->
inputsData
.
empty
())
CV_Error
(
Error
::
StsError
,
"Requested set input"
);
if
(
!
impl
->
netWasAllocated
)
impl
->
setUpNet
();
std
::
ostringstream
out
;
std
::
map
<
int
,
LayerData
>&
map
=
impl
->
layers
;
int
prefBackend
=
impl
->
preferableBackend
;
...
...
modules/dnn/test/test_misc.cpp
浏览文件 @
12fdaf89
...
...
@@ -78,6 +78,26 @@ TEST(readNet, Regression)
EXPECT_FALSE
(
net
.
empty
());
}
typedef
testing
::
TestWithParam
<
tuple
<
Backend
,
Target
>
>
dump
;
TEST_P
(
dump
,
Regression
)
{
const
int
backend
=
get
<
0
>
(
GetParam
());
const
int
target
=
get
<
1
>
(
GetParam
());
Net
net
=
readNet
(
findDataFile
(
"dnn/squeezenet_v1.1.prototxt"
),
findDataFile
(
"dnn/squeezenet_v1.1.caffemodel"
,
false
));
int
size
[]
=
{
1
,
3
,
227
,
227
};
Mat
input
=
cv
::
Mat
::
ones
(
4
,
size
,
CV_32F
);
net
.
setInput
(
input
);
net
.
setPreferableBackend
(
backend
);
net
.
setPreferableTarget
(
target
);
EXPECT_FALSE
(
net
.
dump
().
empty
());
net
.
forward
();
EXPECT_FALSE
(
net
.
dump
().
empty
());
}
INSTANTIATE_TEST_CASE_P
(
/**/
,
dump
,
dnnBackendsAndTargets
());
class
FirstCustomLayer
CV_FINAL
:
public
Layer
{
public:
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录