Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
s920243400
PaddleDetection
提交
e3f1384d
P
PaddleDetection
项目概览
s920243400
/
PaddleDetection
与 Fork 源项目一致
Fork自
PaddlePaddle / PaddleDetection
通知
2
Star
0
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
P
PaddleDetection
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
未验证
提交
e3f1384d
编写于
5月 14, 2020
作者:
W
wangguanzhong
提交者:
GitHub
5月 14, 2020
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
update doc for ext_op (#667)
上级
ba9646fd
变更
4
隐藏空白更改
内联
并排
Showing
4 changed file
with
20 addition
and
3 deletion
+20
-3
configs/anchor_free/README.md
configs/anchor_free/README.md
+1
-0
ppdet/ext_op/README.md
ppdet/ext_op/README.md
+17
-0
ppdet/ext_op/src/make.sh
ppdet/ext_op/src/make.sh
+0
-2
ppdet/modeling/anchor_heads/corner_head.py
ppdet/modeling/anchor_heads/corner_head.py
+2
-1
未找到文件。
configs/anchor_free/README.md
浏览文件 @
e3f1384d
...
@@ -38,6 +38,7 @@
...
@@ -38,6 +38,7 @@
-
CornerNet-Squeeze中使用ResNet结构的骨干网络时,加入了FPN结构,骨干网络的输出feature map采用FPN中的P3层输出。
-
CornerNet-Squeeze中使用ResNet结构的骨干网络时,加入了FPN结构,骨干网络的输出feature map采用FPN中的P3层输出。
-
\*
CornerNet-Squeeze-dcn-mixup-cosine是基于原版CornerNet-Squeeze优化效果最好的模型,在ResNet的骨干网络基础上增加mixup预处理和使用cosine_decay
-
\*
CornerNet-Squeeze-dcn-mixup-cosine是基于原版CornerNet-Squeeze优化效果最好的模型,在ResNet的骨干网络基础上增加mixup预处理和使用cosine_decay
-
FCOS使用GIoU loss、用location分支预测centerness、左上右下角点偏移量归一化和ground truth中心匹配策略
-
FCOS使用GIoU loss、用location分支预测centerness、左上右下角点偏移量归一化和ground truth中心匹配策略
-
Cornernet-Squeeze模型依赖corner_pooling op,该op在
```ppdet/ext_op```
中编译得到,具体编译方式请参考
[
自定义OP的编译过程
](
../../ppdet/ext_op/README.md
)
## 算法细节
## 算法细节
...
...
ppdet/ext_op/README.md
浏览文件 @
e3f1384d
...
@@ -37,6 +37,23 @@ sh make.sh
...
@@ -37,6 +37,23 @@ sh make.sh
编译自定义OP时会报错找不到
`mkldnn.h`
等文件,可在
`make.sh`
中删除编译命令中的
`-DPADDLE_WITH_MKLDNN`
选项。
编译自定义OP时会报错找不到
`mkldnn.h`
等文件,可在
`make.sh`
中删除编译命令中的
`-DPADDLE_WITH_MKLDNN`
选项。
## 设置环境变量
需要将Paddle的核心库设置到
`LD_LIBRARY_PATH`
里, 先运行下面程序获取路径:
```
import paddle
print(paddle.sysconfig.get_lib())
```
可通过如下方式添加动态库路径:
```
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:`python -c 'import paddle; print(paddle.sysconfig.get_lib())'`
```
## 执行单测
## 执行单测
执行下列单测,确保自定义算子可在网络中正确使用:
执行下列单测,确保自定义算子可在网络中正确使用:
...
...
ppdet/ext_op/src/make.sh
浏览文件 @
e3f1384d
...
@@ -19,5 +19,3 @@ g++ bottom_pool_op.cc bottom_pool_op.cu.o top_pool_op.cc top_pool_op.cu.o right_
...
@@ -19,5 +19,3 @@ g++ bottom_pool_op.cc bottom_pool_op.cu.o top_pool_op.cc top_pool_op.cu.o right_
-L
/usr/local/cuda/lib64
-lpaddle_framework
-lcudart
-L
/usr/local/cuda/lib64
-lpaddle_framework
-lcudart
rm
*
.cu.o
rm
*
.cu.o
export
LD_LIBRARY_PATH
=
$LD_LIBRARY_PATH
:
$lib_dir
ppdet/modeling/anchor_heads/corner_head.py
浏览文件 @
e3f1384d
...
@@ -246,7 +246,8 @@ class CornerHead(object):
...
@@ -246,7 +246,8 @@ class CornerHead(object):
try
:
try
:
import
cornerpool_lib
import
cornerpool_lib
except
:
except
:
logger
.
error
(
"cornerpool_lib not found, compile in ext_op at first"
)
logger
.
error
(
"cornerpool_lib not found, compile in ppdet/ext_op at first"
)
self
.
train_batch_size
=
train_batch_size
self
.
train_batch_size
=
train_batch_size
self
.
test_batch_size
=
test_batch_size
self
.
test_batch_size
=
test_batch_size
self
.
num_classes
=
num_classes
self
.
num_classes
=
num_classes
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录