Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
PaddlePaddle
PaddleDetection
提交
59613880
P
PaddleDetection
项目概览
PaddlePaddle
/
PaddleDetection
大约 1 年 前同步成功
通知
695
Star
11112
Fork
2696
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
184
列表
看板
标记
里程碑
合并请求
40
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
P
PaddleDetection
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
184
Issue
184
列表
看板
标记
里程碑
合并请求
40
合并请求
40
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
未验证
提交
59613880
编写于
7月 24, 2020
作者:
K
Kaipeng Deng
提交者:
GitHub
7月 24, 2020
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
add COCO test-dev eval config and doc (#1099)
* add COCO test-dev eval config and doc
上级
f8f73fec
变更
4
显示空白变更内容
内联
并排
Showing
4 changed file
with
178 addition
and
14 deletion
+178
-14
configs/ppyolo/README.md
configs/ppyolo/README.md
+17
-3
configs/ppyolo/ppyolo_test.yml
configs/ppyolo/ppyolo_test.yml
+140
-0
ppdet/modeling/losses/iou_loss.py
ppdet/modeling/losses/iou_loss.py
+2
-2
ppdet/modeling/losses/yolo_loss.py
ppdet/modeling/losses/yolo_loss.py
+19
-9
未找到文件。
configs/ppyolo/README.md
浏览文件 @
59613880
...
...
@@ -11,7 +11,7 @@
[
PP-YOLO
](
https://arxiv.org/abs/2007.12099
)
的PaddleDetection优化和改进的YOLOv3的模型,其精度(COCO数据集mAP)和推理速度均优于
[
YOLOv4
](
https://arxiv.org/abs/2004.10934
)
模型,要求使用PaddlePaddle 1.8.4(2020年8月中旬发布)或适当的
[
develop版本
](
https://www.paddlepaddle.org.cn/documentation/docs/zh/install/Tables.html#whl-dev
)
。
PP-YOLO在
[
COCO
](
http://cocodataset.org
)
test2019数据集上精度达到45.2%,在单卡V100上FP32推理速度为72.9 FPS, V100上开启TensorRT下FP16推理速度为155.6 FPS。
PP-YOLO在
[
COCO
](
http://cocodataset.org
)
test
-dev
2019数据集上精度达到45.2%,在单卡V100上FP32推理速度为72.9 FPS, V100上开启TensorRT下FP16推理速度为155.6 FPS。
<div
align=
"center"
>
<img
src=
"../../docs/images/ppyolo_map_fps.png"
width=
500
/>
...
...
@@ -45,7 +45,7 @@ PP-YOLO从如下方面优化和提升YOLOv3模型的精度和速度:
**注意:**
-
PP-YOLO模型使用COCO数据集中train2017作为训练集,使用test2019左右测试集。
-
PP-YOLO模型使用COCO数据集中train2017作为训练集,使用test
-dev
2019左右测试集。
-
PP-YOLO模型训练过程中使用8GPU,每GPU batch size为24进行训练,如训练GPU数和batch size不使用上述配置,须参考
[
FAQ
](
../../docs/FAQ.md
)
调整学习率和迭代次数。
-
PP-YOLO模型推理速度测试采用单卡V100,batch size=1进行测试,使用CUDA 10.2, CUDNN 7.5.1,TensorRT推理速度测试使用TensorRT 5.1.2.2。
-
PP-YOLO模型推理速度测试数据为使用
`tools/export_model.py`
脚本导出模型后,使用
`deploy/python/infer.py`
脚本中的
`--run_benchnark`
参数使用Paddle预测库进行推理速度benchmark测试结果, 且测试的均为不包含数据预处理和模型输出后处理(NMS)的数据(与
[
YOLOv4(AlexyAB)
](
https://github.com/AlexeyAB/darknet
)
测试方法一致)。
...
...
@@ -66,7 +66,7 @@ CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7 python tools/train.py -c configs/ppyolo/ppy
### 2. 评估
使用单GPU通过如下命令一键式评估模型效果
使用单GPU通过如下命令一键式评估模型
在COCO val2017数据集
效果
```
bash
# 使用PaddleDetection发布的权重
...
...
@@ -76,6 +76,20 @@ CUDA_VISIBLE_DEVICES=0 python tools/eval.py -c configs/ppyolo/ppyolo.yml -o weig
CUDA_VISIBLE_DEVICES
=
0 python tools/eval.py
-c
configs/ppyolo/ppyolo.yml
-o
weights
=
output/ppyolo/best_model
```
我们提供了
`configs/ppyolo/ppyolo_test.yml`
用于评估COCO test-dev2019数据集的效果,评估COCO test-dev2019数据集的效果须先从
[
COCO数据集下载页
](
https://cocodataset.org/#download
)
下载test-dev2019数据集,解压到
`configs/ppyolo/ppyolo_test.yml`
中
`EvalReader.dataset`
中配置的路径,并使用如下命令进行评估
```
bash
# 使用PaddleDetection发布的权重
CUDA_VISIBLE_DEVICES
=
0 python tools/eval.py
-c
configs/ppyolo/ppyolo_test.yml
-o
weights
=
https://paddlemodels.bj.bcebos.com/object_detection/ppyolo.pdparams
# 使用训练保存的checkpoint
CUDA_VISIBLE_DEVICES
=
0 python tools/eval.py
-c
configs/ppyolo/ppyolo_test.yml
-o
weights
=
output/ppyolo/best_model
```
评估结果保存于
`bbox.json`
中,将其压缩为zip包后通过
[
COCO数据集评估页
](
https://competitions.codalab.org/competitions/20794#participate
)
提交评估。
**注意:**
`configs/ppyolo/ppyolo_test.yml`
仅用于评估COCO test-dev数据集,不用于训练和评估COCO val2017数据集。
### 3. 推理
使用单GPU通过如下命令一键式推理图像,通过
`--infer_img`
指定图像路径,或通过
`--infer_dir`
指定目录并推理目录下所有图像
...
...
configs/ppyolo/ppyolo_test.yml
0 → 100644
浏览文件 @
59613880
# NOTE: this config file is only used for evaluation on COCO test2019 set,
# for training or evaluationg on COCO val2017, please use ppyolo.yml
architecture
:
YOLOv3
use_gpu
:
true
max_iters
:
500000
log_smooth_window
:
100
log_iter
:
100
save_dir
:
output
snapshot_iter
:
10000
metric
:
COCO
pretrain_weights
:
https://paddle-imagenet-models-name.bj.bcebos.com/ResNet50_vd_ssld_pretrained.tar
weights
:
output/ppyolo/model_final
num_classes
:
80
use_fine_grained_loss
:
true
use_ema
:
true
ema_decay
:
0.9998
save_prediction_only
:
True
YOLOv3
:
backbone
:
ResNet
yolo_head
:
YOLOv3Head
use_fine_grained_loss
:
true
ResNet
:
norm_type
:
sync_bn
freeze_at
:
0
freeze_norm
:
false
norm_decay
:
0.
depth
:
50
feature_maps
:
[
3
,
4
,
5
]
variant
:
d
dcn_v2_stages
:
[
5
]
YOLOv3Head
:
anchor_masks
:
[[
6
,
7
,
8
],
[
3
,
4
,
5
],
[
0
,
1
,
2
]]
anchors
:
[[
10
,
13
],
[
16
,
30
],
[
33
,
23
],
[
30
,
61
],
[
62
,
45
],
[
59
,
119
],
[
116
,
90
],
[
156
,
198
],
[
373
,
326
]]
norm_decay
:
0.
coord_conv
:
true
iou_aware
:
true
iou_aware_factor
:
0.4
scale_x_y
:
1.05
spp
:
true
yolo_loss
:
YOLOv3Loss
nms
:
MatrixNMS
drop_block
:
true
YOLOv3Loss
:
batch_size
:
24
ignore_thresh
:
0.7
scale_x_y
:
1.05
label_smooth
:
false
use_fine_grained_loss
:
true
iou_loss
:
IouLoss
iou_aware_loss
:
IouAwareLoss
IouLoss
:
loss_weight
:
2.5
max_height
:
608
max_width
:
608
IouAwareLoss
:
loss_weight
:
1.0
max_height
:
608
max_width
:
608
MatrixNMS
:
background_label
:
-1
keep_top_k
:
100
normalized
:
false
score_threshold
:
0.01
post_threshold
:
0.01
LearningRate
:
base_lr
:
0.00333
schedulers
:
-
!PiecewiseDecay
gamma
:
0.1
milestones
:
-
400000
-
450000
-
!LinearWarmup
start_factor
:
0.
steps
:
4000
OptimizerBuilder
:
optimizer
:
momentum
:
0.9
type
:
Momentum
regularizer
:
factor
:
0.0005
type
:
L2
_READER_
:
'
ppyolo_reader.yml'
EvalReader
:
inputs_def
:
fields
:
[
'
image'
,
'
im_size'
,
'
im_id'
]
num_max_boxes
:
90
dataset
:
!COCODataSet
image_dir
:
test2017
anno_path
:
annotations/image_info_test-dev2017.json
dataset_dir
:
dataset/coco
with_background
:
false
sample_transforms
:
-
!DecodeImage
to_rgb
:
True
-
!ResizeImage
target_size
:
608
interp
:
1
-
!NormalizeImage
mean
:
[
0.485
,
0.456
,
0.406
]
std
:
[
0.229
,
0.224
,
0.225
]
is_scale
:
True
is_channel_first
:
false
-
!Permute
to_bgr
:
false
channel_first
:
True
batch_size
:
1
TestReader
:
dataset
:
!ImageFolder
use_default_label
:
true
with_background
:
false
sample_transforms
:
-
!DecodeImage
to_rgb
:
True
-
!ResizeImage
target_size
:
608
interp
:
1
-
!NormalizeImage
mean
:
[
0.485
,
0.456
,
0.406
]
std
:
[
0.229
,
0.224
,
0.225
]
is_scale
:
True
is_channel_first
:
false
-
!Permute
to_bgr
:
false
channel_first
:
True
ppdet/modeling/losses/iou_loss.py
浏览文件 @
59613880
ppdet/modeling/losses/yolo_loss.py
浏览文件 @
59613880
...
...
@@ -91,8 +91,15 @@ class YOLOv3Loss(object):
return
{
'loss'
:
sum
(
losses
)}
def
_get_fine_grained_loss
(
self
,
outputs
,
targets
,
gt_box
,
batch_size
,
num_classes
,
mask_anchors
,
ignore_thresh
,
eps
=
1.e-10
):
def
_get_fine_grained_loss
(
self
,
outputs
,
targets
,
gt_box
,
batch_size
,
num_classes
,
mask_anchors
,
ignore_thresh
,
eps
=
1.e-10
):
"""
Calculate fine grained YOLOv3 loss
...
...
@@ -148,8 +155,10 @@ class YOLOv3Loss(object):
y
,
ty
)
*
tscale_tobj
loss_y
=
fluid
.
layers
.
reduce_sum
(
loss_y
,
dim
=
[
1
,
2
,
3
])
else
:
dx
=
scale_x_y
*
fluid
.
layers
.
sigmoid
(
x
)
-
0.5
*
(
scale_x_y
-
1.0
)
dy
=
scale_x_y
*
fluid
.
layers
.
sigmoid
(
y
)
-
0.5
*
(
scale_x_y
-
1.0
)
dx
=
scale_x_y
*
fluid
.
layers
.
sigmoid
(
x
)
-
0.5
*
(
scale_x_y
-
1.0
)
dy
=
scale_x_y
*
fluid
.
layers
.
sigmoid
(
y
)
-
0.5
*
(
scale_x_y
-
1.0
)
loss_x
=
fluid
.
layers
.
abs
(
dx
-
tx
)
*
tscale_tobj
loss_x
=
fluid
.
layers
.
reduce_sum
(
loss_x
,
dim
=
[
1
,
2
,
3
])
loss_y
=
fluid
.
layers
.
abs
(
dy
-
ty
)
*
tscale_tobj
...
...
@@ -162,7 +171,8 @@ class YOLOv3Loss(object):
loss_h
=
fluid
.
layers
.
reduce_sum
(
loss_h
,
dim
=
[
1
,
2
,
3
])
if
self
.
_iou_loss
is
not
None
:
loss_iou
=
self
.
_iou_loss
(
x
,
y
,
w
,
h
,
tx
,
ty
,
tw
,
th
,
anchors
,
downsample
,
self
.
_batch_size
,
scale_x_y
)
downsample
,
self
.
_batch_size
,
scale_x_y
)
loss_iou
=
loss_iou
*
tscale_tobj
loss_iou
=
fluid
.
layers
.
reduce_sum
(
loss_iou
,
dim
=
[
1
,
2
,
3
])
loss_ious
.
append
(
fluid
.
layers
.
reduce_mean
(
loss_iou
))
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录