Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
PaddlePaddle
X2Paddle
提交
33740b13
X
X2Paddle
项目概览
PaddlePaddle
/
X2Paddle
大约 1 年 前同步成功
通知
328
Star
698
Fork
167
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
26
列表
看板
标记
里程碑
合并请求
4
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
X
X2Paddle
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
26
Issue
26
列表
看板
标记
里程碑
合并请求
4
合并请求
4
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
体验新版 GitCode,发现更多精彩内容 >>
未验证
提交
33740b13
编写于
8月 09, 2019
作者:
S
SunAhong1993
提交者:
GitHub
8月 09, 2019
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
Update add_caffe_custom_layer.md
上级
e8ebd8da
变更
1
隐藏空白更改
内联
并排
Showing
1 changed file
with
46 addition
and
34 deletion
+46
-34
add_caffe_custom_layer.md
add_caffe_custom_layer.md
+46
-34
未找到文件。
add_caffe_custom_layer.md
浏览文件 @
33740b13
## 如何转换Caffe自定义Layer
## 如何转换Caffe自定义Layer
本文档介绍如何将Caffe自定义Layer转换为PaddlePaddle模型中的对应实现, 用户可根据自己需要,添加代码实现自定义层,从而支持模型的完整转换。
本文档介绍如何将Caffe自定义Layer转换为PaddlePaddle模型中的对应实现, 用户可根据自己需要,添加代码实现自定义层,从而支持模型的完整转换。
***步骤一 下载代码**
*
此处涉及修改源码,应先卸载x2paddle,并且下载源码,主要有以下两步完成:
```
pip uninstall x2paddle
pip install git+https://github.com/PaddlePaddle/X2Paddle.git@develop
```
***步骤二 编译caffe.proto**
*
该步骤依赖protobuf编译器,其安装过程有以下两种方式:
> 选择一:pip install protobuf
> 选择二:使用[官方源码](https://github.com/protocolbuffers/protobuf)进行编译
***步骤一 编译caffe.proto**
*
使用脚本./tools/compile.sh将caffe.proto(包含所需的自定义Layer信息)编译成我们所需的目标语言(Python)
使用脚本./tools/compile.sh将caffe.proto(包含所需的自定义Layer信息)编译成我们所需的目标语言(Python)
使用方式:
使用方式:
```
```
...
@@ -10,53 +20,55 @@ bash ./toos/compile.sh /home/root/caffe/src/caffe/proto
...
@@ -10,53 +20,55 @@ bash ./toos/compile.sh /home/root/caffe/src/caffe/proto
# /home/root/caffe/src/caffe/proto为caffe.proto的存放路径,生成的caffe_pb2.py也将保存在该路径下
# /home/root/caffe/src/caffe/proto为caffe.proto的存放路径,生成的caffe_pb2.py也将保存在该路径下
```
```
***步骤二 添加自定义Layer的实现代码**
*
***步骤三 添加自定义Layer的实现代码**
*
-
进入./x2paddle/op_mapper/caffe_custom_layer,创建实现代码的文件,例如mylayer.py
-
进入./x2paddle/op_mapper/caffe_custom_layer,创建.py文件,例如mylayer.py
-
仿照./x2paddle/op_mapper/caffe_custom_layer中的其他文件,在mylayer.py中主要实现3个函数:
-
仿照./x2paddle/op_mapper/caffe_custom_layer中的其他文件,在mylayer.py中主要需要实现3个函数,下面以roipooling.py为例分析代码:
1.
`def mylayer_shape(input_shape, ...)`
1.
`def roipooling_shape(input_shape, pooled_w=None, pooled_h=None)`
参数:
| 参数 | 类型 | 说明 |
1.
input_shape(list):其中每个元素代表该层每个输入数据的shape,为必须传入的参数
| :---------: | :--: | :---------: |
2.
pooled_w(int):代表ROI Pooling的kernel的宽,其命名与.prototxt中roi_pooling_param中的key一致
| input_shape | list | 每个元素代表该层每个输入数据的shape |
3.
pooled_h(int):代表ROI Pooling的kernel的高,其命名与.prototxt中roi_pooling_param中的key一致
| 其余 | 默认为None | 命名为Caffe模型的model.prototxt中mylayer_param中每个参数的名字 |
功能:计算出mylayer的输出shape
返回:一个list,其中每个元素代表每个输出数据的shape
2.
`def mylayer_layer(inputs, input_shape=None, name=None, ...)`
功能:计算出进行ROI Pooling后的shape
返回:一个list,其中每个元素代表每个输出数据的shape,由于ROI Pooling的输出数据只有一个,所以其list长度为1
| 参数 | 类型 | 说明 |
| :---------: | :--: | :---------: |
2.
`def roipooling_layer(inputs, input_shape=None, name=None, pooled_w=None, pooled_h=None, spatial_scale=None)`
| inputs | list | 每个元素代表该层每个输入数据 |
| input_shape | list(默认为None) | 每个元素代表该层每个输入数据的shape |
| name | str(默认为None) | mylayer的名字 |
| 其余 | 默认为None | 命名为Caffe模型的model.prototxt中mylayer_param中每个参数的名字 |
功能:运用PaddlePaddle完成组网来实现
`mylayer`
的功能
参数:
返回:一个Variable或Tensor,为组网后的结果
1.
inputs(list):其中每个元素代表该层每个输入数据,为必须传入的参数
2.
input_shape(list):其中每个元素代表该层每个输入数据的shape,为必须传入的参数
3.
name(str):ROI Pooling层的名字,为必须传入的参数
4.
pooled_w(int):代表ROI Pooling的kernel的宽,其命名与.prototxt中roi_pooling_param中的key一致
5.
pooled_h(int):代表ROI Pooling的kernel的高,其命名与.prototxt中roi_pooling_param中的key一致
6.
spatial_scale(float):用于将ROI坐标从输入比例转换为池化时使用的比例,其命名与.prototxt中roi_pooling_param中的key一致
功能:运用PaddlePaddle完成组网来实现
`roipooling_layer`
的功能
返回:一个Variable,为组网后的结果
3.
`def mylayer_weights(name, data=None)`
3.
`def roipooling_weights(name, data=None)`
| 参数 | 类型 | 说明 |
参数:
| :---------: | :--: | :---------: |
1. name(str):ROI Pooling层的名字,为必须传入的参数
| name | str | mylayer的名字 |
2. data(list):由Caffe模型.caffemodel获得的关于roipooling的参数,roipooling的参数为None
| data | list(默认为None) | 由Caffe模型的model.caffemodel获得的关于mylayer的参数 |
功能:为每个参数(例如kernel、bias等)命名;同时,若Caffe中该层参数与PaddlePaddle中参数的格式不一致,则变换操作也在该函数中实现。
功能:为每个参数(例如kernel、bias等)命名;同时,若Caffe中该层参数与PaddlePaddle中参数的格式不一致,则变换操作也在该函数中实现。
返回:一个list,包含每个参数的名字。
返回:一个list,包含每个参数的名字。
-
在
mylayer.py中注册
`mylayer
`
,主要运用下述代码实现:
-
在
roipooling.py中注册
`roipooling
`
,主要运用下述代码实现:
```
```
register(kind='
Mylayer', shape=mylayer_shape, layer=mylayer_layer, weights=mylayer
_weights)
register(kind='
ROIPooling', shape=roipooling_shape, layer=roipooling_layer, weights=roipooling
_weights)
# kind为在model.prototxt中
mylayer
的type
# kind为在model.prototxt中
roipooling
的type
```
```
-
在./x2paddle/op_mapper/caffe_custom_layer/
\_\_
init
\_\_
.py中引入该层的使用
-
在./x2paddle/op_mapper/caffe_custom_layer/
\_\_
init
\_\_
.py中引入该层的使用
```
```
from . import
mylayer
from . import
roipooling
```
```
***步骤三 运行转换代码**
*
***步骤三 运行转换代码**
*
```
```
# 在X2Paddle目录下安装x2paddle
python setup.py install
# 运行转换代码
x2paddle --framework=caffe
x2paddle --framework=caffe
--prototxt=deploy.proto
--prototxt=deploy.proto
--weight=deploy.caffemodel
--weight=deploy.caffemodel
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录