Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
s920243400
PaddleDetection
提交
20303e69
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看板
体验新版 GitCode,发现更多精彩内容 >>
未验证
提交
20303e69
编写于
5月 01, 2020
作者:
W
whs
提交者:
GitHub
5月 01, 2020
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
Fix eval run in pruning (#576)
上级
1a485e12
变更
3
显示空白变更内容
内联
并排
Showing
3 changed file
with
6 addition
and
3 deletion
+6
-3
ppdet/modeling/anchor_heads/rpn_head.py
ppdet/modeling/anchor_heads/rpn_head.py
+2
-2
ppdet/modeling/tests/test_architectures.py
ppdet/modeling/tests/test_architectures.py
+3
-0
slim/prune/prune.py
slim/prune/prune.py
+1
-1
未找到文件。
ppdet/modeling/anchor_heads/rpn_head.py
浏览文件 @
20303e69
...
@@ -169,7 +169,7 @@ class RPNHead(object):
...
@@ -169,7 +169,7 @@ class RPNHead(object):
rpn_cls_prob
=
fluid
.
layers
.
transpose
(
rpn_cls_prob
=
fluid
.
layers
.
transpose
(
rpn_cls_prob
,
perm
=
[
0
,
3
,
1
,
2
])
rpn_cls_prob
,
perm
=
[
0
,
3
,
1
,
2
])
prop_op
=
self
.
train_proposal
if
mode
==
'train'
else
self
.
test_proposal
prop_op
=
self
.
train_proposal
if
mode
==
'train'
else
self
.
test_proposal
rpn_rois
,
rpn_roi_probs
=
prop_op
(
rpn_rois
,
rpn_roi_probs
,
_
=
prop_op
(
scores
=
rpn_cls_prob
,
scores
=
rpn_cls_prob
,
bbox_deltas
=
rpn_bbox_pred
,
bbox_deltas
=
rpn_bbox_pred
,
im_info
=
im_info
,
im_info
=
im_info
,
...
@@ -430,7 +430,7 @@ class FPNRPNHead(RPNHead):
...
@@ -430,7 +430,7 @@ class FPNRPNHead(RPNHead):
rpn_cls_prob_fpn
,
shape
=
(
0
,
0
,
0
,
-
1
))
rpn_cls_prob_fpn
,
shape
=
(
0
,
0
,
0
,
-
1
))
rpn_cls_prob_fpn
=
fluid
.
layers
.
transpose
(
rpn_cls_prob_fpn
=
fluid
.
layers
.
transpose
(
rpn_cls_prob_fpn
,
perm
=
[
0
,
3
,
1
,
2
])
rpn_cls_prob_fpn
,
perm
=
[
0
,
3
,
1
,
2
])
rpn_rois_fpn
,
rpn_roi_prob_fpn
=
prop_op
(
rpn_rois_fpn
,
rpn_roi_prob_fpn
,
_
=
prop_op
(
scores
=
rpn_cls_prob_fpn
,
scores
=
rpn_cls_prob_fpn
,
bbox_deltas
=
rpn_bbox_pred_fpn
,
bbox_deltas
=
rpn_bbox_pred_fpn
,
im_info
=
im_info
,
im_info
=
im_info
,
...
...
ppdet/modeling/tests/test_architectures.py
浏览文件 @
20303e69
...
@@ -56,6 +56,9 @@ class TestMaskRCNN(TestFasterRCNN):
...
@@ -56,6 +56,9 @@ class TestMaskRCNN(TestFasterRCNN):
self
.
cfg_file
=
'configs/mask_rcnn_r50_1x.yml'
self
.
cfg_file
=
'configs/mask_rcnn_r50_1x.yml'
@
unittest
.
skip
(
reason
=
"It should be fixed to adapt https://github.com/PaddlePaddle/Paddle/pull/23797"
)
class
TestCascadeRCNN
(
TestFasterRCNN
):
class
TestCascadeRCNN
(
TestFasterRCNN
):
def
set_config
(
self
):
def
set_config
(
self
):
self
.
cfg_file
=
'configs/cascade_rcnn_r50_fpn_1x.yml'
self
.
cfg_file
=
'configs/cascade_rcnn_r50_fpn_1x.yml'
...
...
slim/prune/prune.py
浏览文件 @
20303e69
...
@@ -256,7 +256,7 @@ def main():
...
@@ -256,7 +256,7 @@ def main():
if
FLAGS
.
eval
:
if
FLAGS
.
eval
:
# evaluation
# evaluation
results
=
eval_run
(
exe
,
compiled_eval_prog
,
eval_loader
,
eval_keys
,
results
=
eval_run
(
exe
,
compiled_eval_prog
,
eval_loader
,
eval_keys
,
eval_values
,
eval_cls
)
eval_values
,
eval_cls
,
cfg
)
resolution
=
None
resolution
=
None
if
'mask'
in
results
[
0
]:
if
'mask'
in
results
[
0
]:
resolution
=
model
.
mask_head
.
resolution
resolution
=
model
.
mask_head
.
resolution
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录