Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
wangkx1
siton-paddleyolo
Commits
522a602f
Commit
522a602f
authored
Jul 22, 2024
by
wangkx1
Browse files
siton bug
parent
abb99c90
Changes
468
Hide whitespace changes
Inline
Side-by-side
Showing
20 changed files
with
830 additions
and
0 deletions
+830
-0
configs/ppyoloe/application/ppyoloe_plus_crn_m_80e_sku110k.yml
...gs/ppyoloe/application/ppyoloe_plus_crn_m_80e_sku110k.yml
+127
-0
configs/ppyoloe/application/ppyoloe_plus_crn_s_80e_sku110k.yml
...gs/ppyoloe/application/ppyoloe_plus_crn_s_80e_sku110k.yml
+127
-0
configs/ppyoloe/application/ppyoloe_plus_crn_x_80e_sku110k.yml
...gs/ppyoloe/application/ppyoloe_plus_crn_x_80e_sku110k.yml
+127
-0
configs/ppyoloe/distill/README.md
configs/ppyoloe/distill/README.md
+46
-0
configs/ppyoloe/distill/ppyoloe_plus_crn_l_80e_coco_distill.yml
...s/ppyoloe/distill/ppyoloe_plus_crn_l_80e_coco_distill.yml
+39
-0
configs/ppyoloe/distill/ppyoloe_plus_crn_m_80e_coco_distill.yml
...s/ppyoloe/distill/ppyoloe_plus_crn_m_80e_coco_distill.yml
+39
-0
configs/ppyoloe/distill/ppyoloe_plus_crn_s_80e_coco_distill.yml
...s/ppyoloe/distill/ppyoloe_plus_crn_s_80e_coco_distill.yml
+39
-0
configs/ppyoloe/objects365/README_cn.md
configs/ppyoloe/objects365/README_cn.md
+23
-0
configs/ppyoloe/objects365/ppyoloe_plus_crn_l_60e_objects365.yml
.../ppyoloe/objects365/ppyoloe_plus_crn_l_60e_objects365.yml
+21
-0
configs/ppyoloe/objects365/ppyoloe_plus_crn_m_60e_objects365.yml
.../ppyoloe/objects365/ppyoloe_plus_crn_m_60e_objects365.yml
+21
-0
configs/ppyoloe/objects365/ppyoloe_plus_crn_s_60e_objects365.yml
.../ppyoloe/objects365/ppyoloe_plus_crn_s_60e_objects365.yml
+21
-0
configs/ppyoloe/objects365/ppyoloe_plus_crn_x_60e_objects365.yml
.../ppyoloe/objects365/ppyoloe_plus_crn_x_60e_objects365.yml
+21
-0
configs/ppyoloe/ppyoloe_crn_l_300e_coco.yml
configs/ppyoloe/ppyoloe_crn_l_300e_coco.yml
+15
-0
configs/ppyoloe/ppyoloe_crn_l_36e_coco_xpu.yml
configs/ppyoloe/ppyoloe_crn_l_36e_coco_xpu.yml
+71
-0
configs/ppyoloe/ppyoloe_crn_m_300e_coco.yml
configs/ppyoloe/ppyoloe_crn_m_300e_coco.yml
+15
-0
configs/ppyoloe/ppyoloe_crn_s_300e_coco.yml
configs/ppyoloe/ppyoloe_crn_s_300e_coco.yml
+15
-0
configs/ppyoloe/ppyoloe_crn_s_400e_coco.yml
configs/ppyoloe/ppyoloe_crn_s_400e_coco.yml
+18
-0
configs/ppyoloe/ppyoloe_crn_x_300e_coco.yml
configs/ppyoloe/ppyoloe_crn_x_300e_coco.yml
+15
-0
configs/ppyoloe/ppyoloe_plus_crn_l_80e_coco.yml
configs/ppyoloe/ppyoloe_plus_crn_l_80e_coco.yml
+15
-0
configs/ppyoloe/ppyoloe_plus_crn_m_80e_coco.yml
configs/ppyoloe/ppyoloe_plus_crn_m_80e_coco.yml
+15
-0
No files found.
configs/ppyoloe/application/ppyoloe_plus_crn_m_80e_sku110k.yml
0 → 100644
View file @
522a602f
_BASE_
:
[
'
./_base_/sku110k.yml'
,
'
../../runtime.yml'
]
log_iter
:
10
snapshot_epoch
:
20
weights
:
output/ppyoloe_plus_crn_s_80e_coco/model_final
pretrain_weights
:
https://bj.bcebos.com/v1/paddledet/models/pretrained/ppyoloe_crn_m_obj365_pretrained.pdparams
depth_mult
:
0.67
width_mult
:
0.75
# arch
architecture
:
YOLOv3
norm_type
:
sync_bn
use_ema
:
true
ema_decay
:
0.9998
custom_black_list
:
[
'
reduce_mean'
]
YOLOv3
:
backbone
:
CSPResNet
neck
:
CustomCSPPAN
yolo_head
:
PPYOLOEHead
post_process
:
~
CSPResNet
:
layers
:
[
3
,
6
,
6
,
3
]
channels
:
[
64
,
128
,
256
,
512
,
1024
]
return_idx
:
[
1
,
2
,
3
]
use_large_stem
:
True
use_alpha
:
True
CustomCSPPAN
:
out_channels
:
[
768
,
384
,
192
]
stage_num
:
1
block_num
:
3
act
:
'
swish'
spp
:
true
use_alpha
:
True
PPYOLOEHead
:
fpn_strides
:
[
32
,
16
,
8
]
grid_cell_scale
:
5.0
grid_cell_offset
:
0.5
static_assigner_epoch
:
-1
use_varifocal_loss
:
True
loss_weight
:
{
class
:
1.0
,
iou
:
2.5
,
dfl
:
0.5
}
static_assigner
:
name
:
ATSSAssigner
topk
:
9
assigner
:
name
:
TaskAlignedAssigner
topk
:
13
alpha
:
1.0
beta
:
6.0
nms
:
name
:
MultiClassNMS
nms_top_k
:
3000
keep_top_k
:
1000
score_threshold
:
0.01
nms_threshold
:
0.7
# reader
worker_num
:
8
eval_height
:
&eval_height
960
eval_width
:
&eval_width
960
eval_size
:
&eval_size
[
*eval_height
,
*eval_width
]
TrainReader
:
sample_transforms
:
-
Decode
:
{}
-
Resize
:
{
target_size
:
[
3000
,
1800
],
keep_ratio
:
True
,
interp
:
2
}
-
RandomDistort
:
{}
-
RandomCrop
:
{}
-
RandomFlip
:
{}
batch_transforms
:
-
BatchRandomResize
:
{
target_size
:
[
480
,
512
,
544
,
576
,
608
,
640
,
672
,
704
,
736
,
768
,
800
,
832
,
864
,
896
,
928
,
960
,
992
,
1024
,
1056
,
1088
,
1120
,
1152
],
random_size
:
True
,
random_interp
:
True
,
keep_ratio
:
False
}
-
NormalizeImage
:
{
mean
:
[
0.
,
0.
,
0.
],
std
:
[
1.
,
1.
,
1.
],
norm_type
:
none
}
-
Permute
:
{}
-
PadGT
:
{}
batch_size
:
8
shuffle
:
true
drop_last
:
true
use_shared_memory
:
true
collate_batch
:
true
EvalReader
:
sample_transforms
:
-
Decode
:
{}
-
Resize
:
{
target_size
:
*eval_size
,
keep_ratio
:
False
,
interp
:
2
}
-
NormalizeImage
:
{
mean
:
[
0.
,
0.
,
0.
],
std
:
[
1.
,
1.
,
1.
],
norm_type
:
none
}
-
Permute
:
{}
batch_size
:
2
TestReader
:
inputs_def
:
image_shape
:
[
3
,
*eval_height
,
*eval_width
]
sample_transforms
:
-
Decode
:
{}
-
Resize
:
{
target_size
:
*eval_size
,
keep_ratio
:
False
,
interp
:
2
}
-
NormalizeImage
:
{
mean
:
[
0.
,
0.
,
0.
],
std
:
[
1.
,
1.
,
1.
],
norm_type
:
none
}
-
Permute
:
{}
batch_size
:
1
# optimizer
epoch
:
80
LearningRate
:
base_lr
:
0.004
schedulers
:
-
!CosineDecay
max_epochs
:
96
-
!LinearWarmup
start_factor
:
0.
epochs
:
5
OptimizerBuilder
:
optimizer
:
momentum
:
0.9
type
:
Momentum
regularizer
:
factor
:
0.0005
type
:
L2
configs/ppyoloe/application/ppyoloe_plus_crn_s_80e_sku110k.yml
0 → 100644
View file @
522a602f
_BASE_
:
[
'
./_base_/sku110k.yml'
,
'
../../runtime.yml'
]
log_iter
:
10
snapshot_epoch
:
20
weights
:
output/ppyoloe_plus_crn_s_80e_coco/model_final
pretrain_weights
:
https://bj.bcebos.com/v1/paddledet/models/pretrained/ppyoloe_crn_s_obj365_pretrained.pdparams
depth_mult
:
0.33
width_mult
:
0.50
# arch
architecture
:
YOLOv3
norm_type
:
sync_bn
use_ema
:
true
ema_decay
:
0.9998
custom_black_list
:
[
'
reduce_mean'
]
YOLOv3
:
backbone
:
CSPResNet
neck
:
CustomCSPPAN
yolo_head
:
PPYOLOEHead
post_process
:
~
CSPResNet
:
layers
:
[
3
,
6
,
6
,
3
]
channels
:
[
64
,
128
,
256
,
512
,
1024
]
return_idx
:
[
1
,
2
,
3
]
use_large_stem
:
True
use_alpha
:
True
CustomCSPPAN
:
out_channels
:
[
768
,
384
,
192
]
stage_num
:
1
block_num
:
3
act
:
'
swish'
spp
:
true
use_alpha
:
True
PPYOLOEHead
:
fpn_strides
:
[
32
,
16
,
8
]
grid_cell_scale
:
5.0
grid_cell_offset
:
0.5
static_assigner_epoch
:
-1
use_varifocal_loss
:
True
loss_weight
:
{
class
:
1.0
,
iou
:
2.5
,
dfl
:
0.5
}
static_assigner
:
name
:
ATSSAssigner
topk
:
9
assigner
:
name
:
TaskAlignedAssigner
topk
:
13
alpha
:
1.0
beta
:
6.0
nms
:
name
:
MultiClassNMS
nms_top_k
:
3000
keep_top_k
:
1000
score_threshold
:
0.01
nms_threshold
:
0.7
# reader
worker_num
:
8
eval_height
:
&eval_height
960
eval_width
:
&eval_width
960
eval_size
:
&eval_size
[
*eval_height
,
*eval_width
]
TrainReader
:
sample_transforms
:
-
Decode
:
{}
-
Resize
:
{
target_size
:
[
3000
,
1800
],
keep_ratio
:
True
,
interp
:
2
}
-
RandomDistort
:
{}
-
RandomCrop
:
{}
-
RandomFlip
:
{}
batch_transforms
:
-
BatchRandomResize
:
{
target_size
:
[
480
,
512
,
544
,
576
,
608
,
640
,
672
,
704
,
736
,
768
,
800
,
832
,
864
,
896
,
928
,
960
,
992
,
1024
,
1056
,
1088
,
1120
,
1152
],
random_size
:
True
,
random_interp
:
True
,
keep_ratio
:
False
}
-
NormalizeImage
:
{
mean
:
[
0.
,
0.
,
0.
],
std
:
[
1.
,
1.
,
1.
],
norm_type
:
none
}
-
Permute
:
{}
-
PadGT
:
{}
batch_size
:
8
shuffle
:
true
drop_last
:
true
use_shared_memory
:
true
collate_batch
:
true
EvalReader
:
sample_transforms
:
-
Decode
:
{}
-
Resize
:
{
target_size
:
*eval_size
,
keep_ratio
:
False
,
interp
:
2
}
-
NormalizeImage
:
{
mean
:
[
0.
,
0.
,
0.
],
std
:
[
1.
,
1.
,
1.
],
norm_type
:
none
}
-
Permute
:
{}
batch_size
:
2
TestReader
:
inputs_def
:
image_shape
:
[
3
,
*eval_height
,
*eval_width
]
sample_transforms
:
-
Decode
:
{}
-
Resize
:
{
target_size
:
*eval_size
,
keep_ratio
:
False
,
interp
:
2
}
-
NormalizeImage
:
{
mean
:
[
0.
,
0.
,
0.
],
std
:
[
1.
,
1.
,
1.
],
norm_type
:
none
}
-
Permute
:
{}
batch_size
:
1
# optimizer
epoch
:
80
LearningRate
:
base_lr
:
0.004
schedulers
:
-
!CosineDecay
max_epochs
:
96
-
!LinearWarmup
start_factor
:
0.
epochs
:
5
OptimizerBuilder
:
optimizer
:
momentum
:
0.9
type
:
Momentum
regularizer
:
factor
:
0.0005
type
:
L2
configs/ppyoloe/application/ppyoloe_plus_crn_x_80e_sku110k.yml
0 → 100644
View file @
522a602f
_BASE_
:
[
'
./_base_/sku110k.yml'
,
'
../../runtime.yml'
]
log_iter
:
10
snapshot_epoch
:
20
weights
:
output/ppyoloe_plus_crn_s_80e_coco/model_final
pretrain_weights
:
https://bj.bcebos.com/v1/paddledet/models/pretrained/ppyoloe_crn_x_obj365_pretrained.pdparams
depth_mult
:
1.33
width_mult
:
1.25
# arch
architecture
:
YOLOv3
norm_type
:
sync_bn
use_ema
:
true
ema_decay
:
0.9998
custom_black_list
:
[
'
reduce_mean'
]
YOLOv3
:
backbone
:
CSPResNet
neck
:
CustomCSPPAN
yolo_head
:
PPYOLOEHead
post_process
:
~
CSPResNet
:
layers
:
[
3
,
6
,
6
,
3
]
channels
:
[
64
,
128
,
256
,
512
,
1024
]
return_idx
:
[
1
,
2
,
3
]
use_large_stem
:
True
use_alpha
:
True
CustomCSPPAN
:
out_channels
:
[
768
,
384
,
192
]
stage_num
:
1
block_num
:
3
act
:
'
swish'
spp
:
true
use_alpha
:
True
PPYOLOEHead
:
fpn_strides
:
[
32
,
16
,
8
]
grid_cell_scale
:
5.0
grid_cell_offset
:
0.5
static_assigner_epoch
:
-1
use_varifocal_loss
:
True
loss_weight
:
{
class
:
1.0
,
iou
:
2.5
,
dfl
:
0.5
}
static_assigner
:
name
:
ATSSAssigner
topk
:
9
assigner
:
name
:
TaskAlignedAssigner
topk
:
13
alpha
:
1.0
beta
:
6.0
nms
:
name
:
MultiClassNMS
nms_top_k
:
3000
keep_top_k
:
1000
score_threshold
:
0.01
nms_threshold
:
0.7
# reader
worker_num
:
8
eval_height
:
&eval_height
960
eval_width
:
&eval_width
960
eval_size
:
&eval_size
[
*eval_height
,
*eval_width
]
TrainReader
:
sample_transforms
:
-
Decode
:
{}
-
Resize
:
{
target_size
:
[
3000
,
1800
],
keep_ratio
:
True
,
interp
:
2
}
-
RandomDistort
:
{}
-
RandomCrop
:
{}
-
RandomFlip
:
{}
batch_transforms
:
-
BatchRandomResize
:
{
target_size
:
[
480
,
512
,
544
,
576
,
608
,
640
,
672
,
704
,
736
,
768
,
800
,
832
,
864
,
896
,
928
,
960
,
992
,
1024
,
1056
,
1088
,
1120
,
1152
],
random_size
:
True
,
random_interp
:
True
,
keep_ratio
:
False
}
-
NormalizeImage
:
{
mean
:
[
0.
,
0.
,
0.
],
std
:
[
1.
,
1.
,
1.
],
norm_type
:
none
}
-
Permute
:
{}
-
PadGT
:
{}
batch_size
:
4
shuffle
:
true
drop_last
:
true
use_shared_memory
:
true
collate_batch
:
true
EvalReader
:
sample_transforms
:
-
Decode
:
{}
-
Resize
:
{
target_size
:
*eval_size
,
keep_ratio
:
False
,
interp
:
2
}
-
NormalizeImage
:
{
mean
:
[
0.
,
0.
,
0.
],
std
:
[
1.
,
1.
,
1.
],
norm_type
:
none
}
-
Permute
:
{}
batch_size
:
2
TestReader
:
inputs_def
:
image_shape
:
[
3
,
*eval_height
,
*eval_width
]
sample_transforms
:
-
Decode
:
{}
-
Resize
:
{
target_size
:
*eval_size
,
keep_ratio
:
False
,
interp
:
2
}
-
NormalizeImage
:
{
mean
:
[
0.
,
0.
,
0.
],
std
:
[
1.
,
1.
,
1.
],
norm_type
:
none
}
-
Permute
:
{}
batch_size
:
1
# optimizer
epoch
:
80
LearningRate
:
base_lr
:
0.002
schedulers
:
-
!CosineDecay
max_epochs
:
96
-
!LinearWarmup
start_factor
:
0.
epochs
:
5
OptimizerBuilder
:
optimizer
:
momentum
:
0.9
type
:
Momentum
regularizer
:
factor
:
0.0005
type
:
L2
configs/ppyoloe/distill/README.md
0 → 100644
View file @
522a602f
# PPYOLOE+ Distillation(PPYOLOE+ 蒸馏)
PaddleDetection提供了对PPYOLOE+ 进行模型蒸馏的方案,结合了logits蒸馏和feature蒸馏。更多蒸馏方案可以查看
[
slim/distill
](
../../slim/distill/
)
。
## 模型库
| 模型 | 方案 | 输入尺寸 | epochs | Box mAP | 配置文件 | 下载链接 |
| ----------------- | ----------- | ------ | :----: | :-----------: | :--------------: | :------------: |
| PP-YOLOE+_x | teacher | 640 | 80e | 54.7 |
[
config
](
../ppyoloe_plus_crn_x_80e_coco.yml
)
|
[
model
](
https://bj.bcebos.com/v1/paddledet/models/ppyoloe_plus_crn_x_80e_coco.pdparams
)
|
| PP-YOLOE+_l | student | 640 | 80e | 52.9 |
[
config
](
../ppyoloe_plus_crn_l_80e_coco.yml
)
|
[
model
](
https://bj.bcebos.com/v1/paddledet/models/ppyoloe_plus_crn_l_80e_coco.pdparams
)
|
| PP-YOLOE+_l | distill | 640 | 80e |
**54.0(+1.1)**
|
[
config
](
./ppyoloe_plus_crn_l_80e_coco_distill.yml
)
,
[
slim_config
](
../../slim/distill/ppyoloe_plus_distill_x_distill_l.yml
)
|
[
model
](
https://bj.bcebos.com/v1/paddledet/models/ppyoloe_plus_crn_l_80e_coco_distill.pdparams
)
|
| PP-YOLOE+_l | teacher | 640 | 80e | 52.9 |
[
config
](
../ppyoloe_plus_crn_l_80e_coco.yml
)
|
[
model
](
https://bj.bcebos.com/v1/paddledet/models/ppyoloe_plus_crn_l_80e_coco.pdparams
)
|
| PP-YOLOE+_m | student | 640 | 80e | 49.8 |
[
config
](
../ppyoloe_plus_crn_m_80e_coco.yml
)
|
[
model
](
https://bj.bcebos.com/v1/paddledet/models/ppyoloe_plus_crn_m_80e_coco.pdparams
)
|
| PP-YOLOE+_m | distill | 640 | 80e |
**51.0(+1.2)**
|
[
config
](
./ppyoloe_plus_crn_m_80e_coco_distill.yml
)
,
[
slim_config
](
../../slim/distill/ppyoloe_plus_distill_l_distill_m.yml
)
|
[
model
](
https://bj.bcebos.com/v1/paddledet/models/ppyoloe_plus_crn_m_80e_coco_distill.pdparams
)
|
## 快速开始
### 训练
```
shell
# 单卡
python tools/train.py
-c
configs/ppyoloe/distill/ppyoloe_plus_crn_l_80e_coco_distill.yml
--slim_config
configs/slim/distill/ppyoloe_plus_distill_x_distill_l.yml
# 多卡
python
-m
paddle.distributed.launch
--log_dir
=
ppyoloe_plus_distill_x_distill_l/
--gpus
0,1,2,3,4,5,6,7 tools/train.py
-c
configs/ppyoloe/distill/ppyoloe_plus_crn_l_80e_coco_distill.yml
--slim_config
configs/slim/distill/ppyoloe_plus_distill_x_distill_l.yml
```
-
`-c`
: 指定模型配置文件,也是student配置文件。
-
`--slim_config`
: 指定压缩策略配置文件,也是teacher配置文件。
### 评估
```
shell
python tools/eval.py
-c
configs/ppyoloe/distill/ppyoloe_plus_crn_l_80e_coco_distill.yml
-o
weights
=
output/ppyoloe_plus_crn_l_80e_coco_distill/model_final.pdparams
```
-
`-c`
: 指定模型配置文件,也是student配置文件。
-
`--slim_config`
: 指定压缩策略配置文件,也是teacher配置文件。
-
`-o weights`
: 指定压缩算法训好的模型路径。
### 测试
```
shell
python tools/infer.py
-c
configs/ppyoloe/distill/ppyoloe_plus_crn_l_80e_coco_distill.yml
-o
weights
=
output/ppyoloe_plus_crn_l_80e_coco_distill/model_final.pdparams
--infer_img
=
demo/000000014439_640x640.jpg
```
-
`-c`
: 指定模型配置文件。
-
`--slim_config`
: 指定压缩策略配置文件。
-
`-o weights`
: 指定压缩算法训好的模型路径。
-
`--infer_img`
: 指定测试图像路径。
configs/ppyoloe/distill/ppyoloe_plus_crn_l_80e_coco_distill.yml
0 → 100644
View file @
522a602f
_BASE_
:
[
'
../ppyoloe_plus_crn_l_80e_coco.yml'
,
]
for_distill
:
True
architecture
:
PPYOLOE
PPYOLOE
:
backbone
:
CSPResNet
neck
:
CustomCSPPAN
yolo_head
:
PPYOLOEHead
post_process
:
~
worker_num
:
4
TrainReader
:
sample_transforms
:
-
Decode
:
{}
-
RandomDistort
:
{}
-
RandomExpand
:
{
fill_value
:
[
123.675
,
116.28
,
103.53
]}
-
RandomCrop
:
{}
-
RandomFlip
:
{}
batch_transforms
:
-
BatchRandomResize
:
{
target_size
:
[
640
],
random_size
:
True
,
random_interp
:
True
,
keep_ratio
:
False
}
-
NormalizeImage
:
{
mean
:
[
0.
,
0.
,
0.
],
std
:
[
1.
,
1.
,
1.
],
norm_type
:
none
}
-
Permute
:
{}
-
PadGT
:
{}
batch_size
:
8
shuffle
:
True
drop_last
:
True
use_shared_memory
:
True
collate_batch
:
True
log_iter
:
100
snapshot_epoch
:
5
weights
:
output/ppyoloe_plus_crn_l_80e_coco_distill/model_final
pretrain_weights
:
https://bj.bcebos.com/v1/paddledet/models/pretrained/ppyoloe_crn_l_obj365_pretrained.pdparams
depth_mult
:
1.0
width_mult
:
1.0
configs/ppyoloe/distill/ppyoloe_plus_crn_m_80e_coco_distill.yml
0 → 100644
View file @
522a602f
_BASE_
:
[
'
../ppyoloe_plus_crn_m_80e_coco.yml'
,
]
for_distill
:
True
architecture
:
PPYOLOE
PPYOLOE
:
backbone
:
CSPResNet
neck
:
CustomCSPPAN
yolo_head
:
PPYOLOEHead
post_process
:
~
worker_num
:
4
TrainReader
:
sample_transforms
:
-
Decode
:
{}
-
RandomDistort
:
{}
-
RandomExpand
:
{
fill_value
:
[
123.675
,
116.28
,
103.53
]}
-
RandomCrop
:
{}
-
RandomFlip
:
{}
batch_transforms
:
-
BatchRandomResize
:
{
target_size
:
[
640
],
random_size
:
True
,
random_interp
:
True
,
keep_ratio
:
False
}
-
NormalizeImage
:
{
mean
:
[
0.
,
0.
,
0.
],
std
:
[
1.
,
1.
,
1.
],
norm_type
:
none
}
-
Permute
:
{}
-
PadGT
:
{}
batch_size
:
8
shuffle
:
True
drop_last
:
True
use_shared_memory
:
True
collate_batch
:
True
log_iter
:
100
snapshot_epoch
:
5
weights
:
output/ppyoloe_plus_crn_m_80e_coco_distill/model_final
pretrain_weights
:
https://bj.bcebos.com/v1/paddledet/models/pretrained/ppyoloe_crn_m_obj365_pretrained.pdparams
depth_mult
:
0.67
width_mult
:
0.75
configs/ppyoloe/distill/ppyoloe_plus_crn_s_80e_coco_distill.yml
0 → 100644
View file @
522a602f
_BASE_
:
[
'
../ppyoloe_plus_crn_s_80e_coco.yml'
,
]
for_distill
:
True
architecture
:
PPYOLOE
PPYOLOE
:
backbone
:
CSPResNet
neck
:
CustomCSPPAN
yolo_head
:
PPYOLOEHead
post_process
:
~
worker_num
:
4
TrainReader
:
sample_transforms
:
-
Decode
:
{}
-
RandomDistort
:
{}
-
RandomExpand
:
{
fill_value
:
[
123.675
,
116.28
,
103.53
]}
-
RandomCrop
:
{}
-
RandomFlip
:
{}
batch_transforms
:
-
BatchRandomResize
:
{
target_size
:
[
640
],
random_size
:
True
,
random_interp
:
True
,
keep_ratio
:
False
}
-
NormalizeImage
:
{
mean
:
[
0.
,
0.
,
0.
],
std
:
[
1.
,
1.
,
1.
],
norm_type
:
none
}
-
Permute
:
{}
-
PadGT
:
{}
batch_size
:
8
shuffle
:
True
drop_last
:
True
use_shared_memory
:
True
collate_batch
:
True
log_iter
:
100
snapshot_epoch
:
5
weights
:
output/ppyoloe_plus_crn_s_80e_coco_distill/model_final
pretrain_weights
:
https://bj.bcebos.com/v1/paddledet/models/pretrained/ppyoloe_crn_s_obj365_pretrained.pdparams
depth_mult
:
0.33
width_mult
:
0.50
configs/ppyoloe/objects365/README_cn.md
0 → 100644
View file @
522a602f
# PP-YOLOE
## 模型库
### Objects365数据集模型库
| 模型 | Epoch | 机器个数 | GPU个数 | 每GPU图片个数 | 骨干网络 | 输入尺寸 | Box AP
<sup>
0.5 | Params(M) | FLOPs(G) | V100 FP32(FPS) | V100 TensorRT FP16(FPS) | 模型下载 | 配置文件 |
|:---------------:|:-----:|:-----------:|:-----------:|:-----------:|:---------:|:----------:|:--------------:|:---------:|:---------:|:-------------:|:-----------------------:| :--------:|:--------:|
| PP-YOLOE+_s | 60 | 3 | 8 | 8 | cspresnet-s | 640 | 18.1 | 7.93 | 17.36 | 208.3 | 333.3 |
[
model
](
https://bj.bcebos.com/v1/paddledet/models/pretrained/ppyoloe_crn_s_obj365_pretrained.pdparams
)
|
[
config
](
./ppyoloe_plus_crn_s_60e_objects365.yml
)
|
| PP-YOLOE+_m | 60 | 4 | 8 | 8 | cspresnet-m | 640 | 25.0 | 23.43 | 49.91 | 123.4 | 208.3 |
[
model
](
https://bj.bcebos.com/v1/paddledet/models/pretrained/ppyoloe_crn_m_obj365_pretrained.pdparams
)
|
[
config
](
./ppyoloe_plus_crn_m_60e_objects365.yml
)
|
| PP-YOLOE+_l | 60 | 3 | 8 | 8 | cspresnet-l | 640 | 30.8 | 52.20 | 110.07 | 78.1 | 149.2 |
[
model
](
https://bj.bcebos.com/v1/paddledet/models/pretrained/ppyoloe_crn_l_obj365_pretrained.pdparams
)
|
[
config
](
./ppyoloe_plus_crn_l_60e_objects365.yml
)
|
| PP-YOLOE+_x | 60 | 4 | 8 | 8 | cspresnet-x | 640 | 32.7 | 98.42 | 206.59 | 45.0 | 95.2 |
[
model
](
https://bj.bcebos.com/v1/paddledet/models/pretrained/ppyoloe_crn_x_obj365_pretrained.pdparams
)
|
[
config
](
./ppyoloe_plus_crn_x_60e_objects365.yml
)
|
**注意:**
-
多机训练细节见
[
文档
](
../../../docs/tutorials/DistributedTraining_cn.md
)
-
Objects365数据集下载请参考
[
objects365官网
](
http://www.objects365.org/overview.html
)
。具体种类列表可下载由PaddleDetection团队整理的
[
objects365_detection_label_list.txt
](
https://bj.bcebos.com/v1/paddledet/data/objects365/objects365_detection_label_list.txt
)
并存放在
`dataset/objects365/`
,每一行即表示第几个种类。inference或导出模型时需要读取到种类数,如果没有标注json文件时,可以进行如下更改
`configs/datasets/objects365_detection.yml`
:
```
TestDataset:
!ImageFolder
# anno_path: annotations/zhiyuan_objv2_val.json
anno_path: objects365_detection_label_list.txt
dataset_dir: dataset/objects365/
```
configs/ppyoloe/objects365/ppyoloe_plus_crn_l_60e_objects365.yml
0 → 100644
View file @
522a602f
_BASE_
:
[
'
../../datasets/objects365_detection.yml'
,
'
../../runtime.yml'
,
'
../_base_/optimizer_60e.yml'
,
'
../_base_/ppyoloe_plus_crn.yml'
,
'
../_base_/ppyoloe_plus_reader.yml'
,
]
log_iter
:
100
snapshot_epoch
:
5
weights
:
output/ppyoloe_plus_crn_l_60e_objects365/model_final
pretrain_weights
:
https://paddledet.bj.bcebos.com/models/pretrained/CSPResNetb_l_pretrained.pdparams
CSPResNet
:
use_alpha
:
False
PPYOLOEHead
:
static_assigner_epoch
:
20
depth_mult
:
1.0
width_mult
:
1.0
configs/ppyoloe/objects365/ppyoloe_plus_crn_m_60e_objects365.yml
0 → 100644
View file @
522a602f
_BASE_
:
[
'
../../datasets/objects365_detection.yml'
,
'
../../runtime.yml'
,
'
../_base_/optimizer_60e.yml'
,
'
../_base_/ppyoloe_plus_crn.yml'
,
'
../_base_/ppyoloe_plus_reader.yml'
,
]
log_iter
:
100
snapshot_epoch
:
5
weights
:
output/ppyoloe_plus_crn_m_60e_objects365/model_final
pretrain_weights
:
https://paddledet.bj.bcebos.com/models/pretrained/CSPResNetb_m_pretrained.pdparams
CSPResNet
:
use_alpha
:
False
PPYOLOEHead
:
static_assigner_epoch
:
20
depth_mult
:
0.67
width_mult
:
0.75
configs/ppyoloe/objects365/ppyoloe_plus_crn_s_60e_objects365.yml
0 → 100644
View file @
522a602f
_BASE_
:
[
'
../../datasets/objects365_detection.yml'
,
'
../../runtime.yml'
,
'
../_base_/optimizer_60e.yml'
,
'
../_base_/ppyoloe_plus_crn.yml'
,
'
../_base_/ppyoloe_plus_reader.yml'
,
]
log_iter
:
100
snapshot_epoch
:
5
weights
:
output/ppyoloe_plus_crn_s_60e_objects365/model_final
pretrain_weights
:
https://paddledet.bj.bcebos.com/models/pretrained/CSPResNetb_s_pretrained.pdparams
CSPResNet
:
use_alpha
:
False
PPYOLOEHead
:
static_assigner_epoch
:
20
depth_mult
:
0.33
width_mult
:
0.50
configs/ppyoloe/objects365/ppyoloe_plus_crn_x_60e_objects365.yml
0 → 100644
View file @
522a602f
_BASE_
:
[
'
../../datasets/objects365_detection.yml'
,
'
../../runtime.yml'
,
'
../_base_/optimizer_60e.yml'
,
'
../_base_/ppyoloe_plus_crn.yml'
,
'
../_base_/ppyoloe_plus_reader.yml'
,
]
log_iter
:
100
snapshot_epoch
:
5
weights
:
output/ppyoloe_plus_crn_x_60e_objects365/model_final
pretrain_weights
:
https://paddledet.bj.bcebos.com/models/pretrained/CSPResNetb_x_pretrained.pdparams
CSPResNet
:
use_alpha
:
False
PPYOLOEHead
:
static_assigner_epoch
:
20
depth_mult
:
1.33
width_mult
:
1.25
configs/ppyoloe/ppyoloe_crn_l_300e_coco.yml
0 → 100644
View file @
522a602f
_BASE_
:
[
'
../datasets/coco_detection.yml'
,
'
../runtime.yml'
,
'
./_base_/optimizer_300e.yml'
,
'
./_base_/ppyoloe_crn.yml'
,
'
./_base_/ppyoloe_reader.yml'
,
]
log_iter
:
100
snapshot_epoch
:
10
weights
:
output/ppyoloe_crn_l_300e_coco/model_final
pretrain_weights
:
https://paddledet.bj.bcebos.com/models/pretrained/CSPResNetb_l_pretrained.pdparams
depth_mult
:
1.0
width_mult
:
1.0
configs/ppyoloe/ppyoloe_crn_l_36e_coco_xpu.yml
0 → 100644
View file @
522a602f
_BASE_
:
[
'
../datasets/coco_detection.yml'
,
'
../runtime.yml'
,
'
./_base_/optimizer_36e_xpu.yml'
,
'
./_base_/ppyoloe_reader.yml'
,
]
# note: these are default values (use_gpu = true and use_xpu = false) for CI.
# set use_gpu = false and use_xpu = true for training.
use_gpu
:
true
use_xpu
:
false
log_iter
:
100
snapshot_epoch
:
1
weights
:
output/ppyoloe_crn_l_36e_coco/model_final
find_unused_parameters
:
True
pretrain_weights
:
https://paddledet.bj.bcebos.com/models/pretrained/CSPResNetb_l_pretrained.pdparams
depth_mult
:
1.0
width_mult
:
1.0
TrainReader
:
batch_size
:
8
architecture
:
YOLOv3
norm_type
:
sync_bn
use_ema
:
true
ema_decay
:
0.9998
ema_black_list
:
[
'
proj_conv.weight'
]
custom_black_list
:
[
'
reduce_mean'
]
YOLOv3
:
backbone
:
CSPResNet
neck
:
CustomCSPPAN
yolo_head
:
PPYOLOEHead
post_process
:
~
CSPResNet
:
layers
:
[
3
,
6
,
6
,
3
]
channels
:
[
64
,
128
,
256
,
512
,
1024
]
return_idx
:
[
1
,
2
,
3
]
use_large_stem
:
True
CustomCSPPAN
:
out_channels
:
[
768
,
384
,
192
]
stage_num
:
1
block_num
:
3
act
:
'
swish'
spp
:
true
PPYOLOEHead
:
fpn_strides
:
[
32
,
16
,
8
]
grid_cell_scale
:
5.0
grid_cell_offset
:
0.5
static_assigner_epoch
:
4
use_varifocal_loss
:
True
loss_weight
:
{
class
:
1.0
,
iou
:
2.5
,
dfl
:
0.5
}
static_assigner
:
name
:
ATSSAssigner
topk
:
9
assigner
:
name
:
TaskAlignedAssigner
topk
:
13
alpha
:
1.0
beta
:
6.0
nms
:
name
:
MultiClassNMS
nms_top_k
:
1000
keep_top_k
:
300
score_threshold
:
0.01
nms_threshold
:
0.7
configs/ppyoloe/ppyoloe_crn_m_300e_coco.yml
0 → 100644
View file @
522a602f
_BASE_
:
[
'
../datasets/coco_detection.yml'
,
'
../runtime.yml'
,
'
./_base_/optimizer_300e.yml'
,
'
./_base_/ppyoloe_crn.yml'
,
'
./_base_/ppyoloe_reader.yml'
,
]
log_iter
:
100
snapshot_epoch
:
10
weights
:
output/ppyoloe_crn_m_300e_coco/model_final
pretrain_weights
:
https://paddledet.bj.bcebos.com/models/pretrained/CSPResNetb_m_pretrained.pdparams
depth_mult
:
0.67
width_mult
:
0.75
configs/ppyoloe/ppyoloe_crn_s_300e_coco.yml
0 → 100644
View file @
522a602f
_BASE_
:
[
'
../datasets/coco_detection.yml'
,
'
../runtime.yml'
,
'
./_base_/optimizer_300e.yml'
,
'
./_base_/ppyoloe_crn.yml'
,
'
./_base_/ppyoloe_reader.yml'
,
]
log_iter
:
100
snapshot_epoch
:
10
weights
:
output/ppyoloe_crn_s_300e_coco/model_final
pretrain_weights
:
https://paddledet.bj.bcebos.com/models/pretrained/CSPResNetb_s_pretrained.pdparams
depth_mult
:
0.33
width_mult
:
0.50
configs/ppyoloe/ppyoloe_crn_s_400e_coco.yml
0 → 100644
View file @
522a602f
_BASE_
:
[
'
../datasets/coco_detection.yml'
,
'
../runtime.yml'
,
'
./_base_/optimizer_400e.yml'
,
'
./_base_/ppyoloe_crn.yml'
,
'
./_base_/ppyoloe_reader.yml'
,
]
log_iter
:
100
snapshot_epoch
:
10
weights
:
output/ppyoloe_crn_s_400e_coco/model_final
pretrain_weights
:
https://paddledet.bj.bcebos.com/models/pretrained/CSPResNetb_s_pretrained.pdparams
depth_mult
:
0.33
width_mult
:
0.50
PPYOLOEHead
:
static_assigner_epoch
:
133
configs/ppyoloe/ppyoloe_crn_x_300e_coco.yml
0 → 100644
View file @
522a602f
_BASE_
:
[
'
../datasets/coco_detection.yml'
,
'
../runtime.yml'
,
'
./_base_/optimizer_300e.yml'
,
'
./_base_/ppyoloe_crn.yml'
,
'
./_base_/ppyoloe_reader.yml'
,
]
log_iter
:
100
snapshot_epoch
:
10
weights
:
output/ppyoloe_crn_x_300e_coco/model_final
pretrain_weights
:
https://paddledet.bj.bcebos.com/models/pretrained/CSPResNetb_x_pretrained.pdparams
depth_mult
:
1.33
width_mult
:
1.25
configs/ppyoloe/ppyoloe_plus_crn_l_80e_coco.yml
0 → 100644
View file @
522a602f
_BASE_
:
[
'
../datasets/coco_detection.yml'
,
'
../runtime.yml'
,
'
./_base_/optimizer_80e.yml'
,
'
./_base_/ppyoloe_plus_crn.yml'
,
'
./_base_/ppyoloe_plus_reader.yml'
,
]
log_iter
:
100
snapshot_epoch
:
5
weights
:
output/ppyoloe_plus_crn_l_80e_coco/model_final
pretrain_weights
:
https://bj.bcebos.com/v1/paddledet/models/pretrained/ppyoloe_crn_l_obj365_pretrained.pdparams
depth_mult
:
1.0
width_mult
:
1.0
configs/ppyoloe/ppyoloe_plus_crn_m_80e_coco.yml
0 → 100644
View file @
522a602f
_BASE_
:
[
'
../datasets/coco_detection.yml'
,
'
../runtime.yml'
,
'
./_base_/optimizer_80e.yml'
,
'
./_base_/ppyoloe_plus_crn.yml'
,
'
./_base_/ppyoloe_plus_reader.yml'
,
]
log_iter
:
100
snapshot_epoch
:
5
weights
:
output/ppyoloe_plus_crn_m_80e_coco/model_final
pretrain_weights
:
https://bj.bcebos.com/v1/paddledet/models/pretrained/ppyoloe_crn_m_obj365_pretrained.pdparams
depth_mult
:
0.67
width_mult
:
0.75
Prev
1
2
3
4
5
6
7
8
9
10
…
24
Next
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment