Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
Greenplum
Opencv
提交
eabf7286
O
Opencv
项目概览
Greenplum
/
Opencv
大约 1 年 前同步成功
通知
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,发现更多精彩内容 >>
提交
eabf7286
编写于
10月 09, 2017
作者:
D
Dmitry Kurtaev
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
PReLU layer from Caffe
上级
bfb12acd
变更
5
显示空白变更内容
内联
并排
Showing
5 changed file
with
18 addition
and
5 deletion
+18
-5
modules/dnn/include/opencv2/dnn/all_layers.hpp
modules/dnn/include/opencv2/dnn/all_layers.hpp
+1
-1
modules/dnn/src/caffe/caffe_importer.cpp
modules/dnn/src/caffe/caffe_importer.cpp
+3
-3
modules/dnn/src/init.cpp
modules/dnn/src/init.cpp
+1
-0
modules/dnn/src/layers/elementwise_layers.cpp
modules/dnn/src/layers/elementwise_layers.cpp
+8
-1
modules/dnn/test/test_layers.cpp
modules/dnn/test/test_layers.cpp
+5
-0
未找到文件。
modules/dnn/include/opencv2/dnn/all_layers.hpp
浏览文件 @
eabf7286
...
...
@@ -422,7 +422,7 @@ CV__DNN_EXPERIMENTAL_NS_BEGIN
class
CV_EXPORTS
ChannelsPReLULayer
:
public
ActivationLayer
{
public:
static
Ptr
<
ChannelsPReLU
Layer
>
create
(
const
LayerParams
&
params
);
static
Ptr
<
Layer
>
create
(
const
LayerParams
&
params
);
};
class
CV_EXPORTS
ELULayer
:
public
ActivationLayer
...
...
modules/dnn/src/caffe/caffe_importer.cpp
浏览文件 @
eabf7286
...
...
@@ -216,7 +216,7 @@ public:
shape
.
push_back
((
int
)
_shape
.
dim
(
i
));
}
else
CV_Error
(
Error
::
StsError
,
"Unknown shape of input blob"
);
shape
.
resize
(
1
,
1
);
// Is a scalar.
}
void
blobFromProto
(
const
caffe
::
BlobProto
&
pbBlob
,
cv
::
Mat
&
dstBlob
)
...
...
@@ -274,9 +274,9 @@ public:
struct
BlobNote
{
BlobNote
(
const
std
::
string
&
_name
,
int
_layerId
,
int
_outNum
)
:
name
(
_name
.
c_str
()
),
layerId
(
_layerId
),
outNum
(
_outNum
)
{}
name
(
_name
),
layerId
(
_layerId
),
outNum
(
_outNum
)
{}
const
char
*
name
;
std
::
string
name
;
int
layerId
,
outNum
;
};
...
...
modules/dnn/src/init.cpp
浏览文件 @
eabf7286
...
...
@@ -97,6 +97,7 @@ void initializeLayerFactory()
CV_DNN_REGISTER_LAYER_CLASS
(
ReLU
,
ReLULayer
);
CV_DNN_REGISTER_LAYER_CLASS
(
ReLU6
,
ReLU6Layer
);
CV_DNN_REGISTER_LAYER_CLASS
(
ChannelsPReLU
,
ChannelsPReLULayer
);
CV_DNN_REGISTER_LAYER_CLASS
(
PReLU
,
ChannelsPReLULayer
);
CV_DNN_REGISTER_LAYER_CLASS
(
Sigmoid
,
SigmoidLayer
);
CV_DNN_REGISTER_LAYER_CLASS
(
TanH
,
TanHLayer
);
CV_DNN_REGISTER_LAYER_CLASS
(
ELU
,
ELULayer
);
...
...
modules/dnn/src/layers/elementwise_layers.cpp
浏览文件 @
eabf7286
...
...
@@ -754,8 +754,15 @@ Ptr<PowerLayer> PowerLayer::create(const LayerParams& params)
return
l
;
}
Ptr
<
ChannelsPReLU
Layer
>
ChannelsPReLULayer
::
create
(
const
LayerParams
&
params
)
Ptr
<
Layer
>
ChannelsPReLULayer
::
create
(
const
LayerParams
&
params
)
{
CV_Assert
(
params
.
blobs
.
size
()
==
1
);
if
(
params
.
blobs
[
0
].
total
()
==
1
)
{
LayerParams
reluParams
=
params
;
reluParams
.
set
(
"negative_slope"
,
params
.
blobs
[
0
].
at
<
float
>
(
0
));
return
ReLULayer
::
create
(
reluParams
);
}
Ptr
<
ChannelsPReLULayer
>
l
(
new
ElementWiseLayer
<
ChannelsPReLUFunctor
>
(
ChannelsPReLUFunctor
(
params
.
blobs
[
0
])));
l
->
setParamsFrom
(
params
);
...
...
modules/dnn/test/test_layers.cpp
浏览文件 @
eabf7286
...
...
@@ -279,6 +279,11 @@ TEST(Layer_Test_Eltwise, Accuracy)
testLayerUsingCaffeModels
(
"layer_eltwise"
);
}
TEST
(
Layer_Test_PReLU
,
Accuracy
)
{
testLayerUsingCaffeModels
(
"layer_prelu"
,
DNN_TARGET_CPU
,
true
);
}
//template<typename XMat>
//static void test_Layer_Concat()
//{
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录