更换文档检测模型
This commit is contained in:
9
paddle_detection/configs/ppyoloe/voc/README_cn.md
Normal file
9
paddle_detection/configs/ppyoloe/voc/README_cn.md
Normal file
@@ -0,0 +1,9 @@
|
||||
# PP-YOLOE
|
||||
|
||||
## 模型库
|
||||
|
||||
### VOC数据集模型库
|
||||
| 模型 | Epoch | GPU个数 | 每GPU图片个数 | 骨干网络 | 输入尺寸 | Box AP<sup>0.5 | Params(M) | FLOPs(G) | V100 FP32(FPS) | V100 TensorRT FP16(FPS) | 模型下载 | 配置文件 |
|
||||
|:---------------:|:-----:|:-----------:|:-----------:|:---------:|:----------:|:--------------:|:---------:|:---------:|:-------------:|:-----------------------:| :-------: |:--------:|
|
||||
| PP-YOLOE+_s | 30 | 8 | 8 | cspresnet-s | 640 | 86.7 | 7.93 | 17.36 | 208.3 | 333.3 | [model](https://paddledet.bj.bcebos.com/models/ppyoloe_plus_crn_s_30e_voc.pdparams) | [config](./ppyoloe_plus_crn_s_30e_voc.yml) |
|
||||
| PP-YOLOE+_l | 30 | 8 | 8 | cspresnet-l | 640 | 89.0 | 52.20 | 110.07 | 78.1 | 149.2 | [model](https://paddledet.bj.bcebos.com/models/ppyoloe_plus_crn_l_30e_voc.pdparams) | [config](./ppyoloe_plus_crn_l_30e_voc.yml) |
|
||||
@@ -0,0 +1,43 @@
|
||||
_BASE_: [
|
||||
'../../datasets/voc.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_30e_voc/model_final
|
||||
|
||||
pretrain_weights: https://bj.bcebos.com/v1/paddledet/models/ppyoloe_plus_crn_l_80e_coco.pdparams
|
||||
depth_mult: 1.0
|
||||
width_mult: 1.0
|
||||
|
||||
|
||||
TrainReader:
|
||||
batch_size: 8 # default 8 gpus, total bs = 64
|
||||
|
||||
EvalReader:
|
||||
batch_size: 4
|
||||
|
||||
|
||||
epoch: 30
|
||||
LearningRate:
|
||||
base_lr: 0.001
|
||||
schedulers:
|
||||
- !CosineDecay
|
||||
max_epochs: 36
|
||||
- !LinearWarmup
|
||||
start_factor: 0.
|
||||
epochs: 1
|
||||
|
||||
|
||||
PPYOLOEHead:
|
||||
static_assigner_epoch: -1
|
||||
nms:
|
||||
name: MultiClassNMS
|
||||
nms_top_k: 1000
|
||||
keep_top_k: 300
|
||||
score_threshold: 0.01
|
||||
nms_threshold: 0.7
|
||||
@@ -0,0 +1,43 @@
|
||||
_BASE_: [
|
||||
'../../datasets/voc.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_s_30e_voc/model_final
|
||||
|
||||
pretrain_weights: https://bj.bcebos.com/v1/paddledet/models/ppyoloe_plus_crn_s_80e_coco.pdparams
|
||||
depth_mult: 0.33
|
||||
width_mult: 0.50
|
||||
|
||||
|
||||
TrainReader:
|
||||
batch_size: 8 # default 8 gpus, total bs = 64
|
||||
|
||||
EvalReader:
|
||||
batch_size: 4
|
||||
|
||||
|
||||
epoch: 30
|
||||
LearningRate:
|
||||
base_lr: 0.001
|
||||
schedulers:
|
||||
- !CosineDecay
|
||||
max_epochs: 36
|
||||
- !LinearWarmup
|
||||
start_factor: 0.
|
||||
epochs: 1
|
||||
|
||||
|
||||
PPYOLOEHead:
|
||||
static_assigner_epoch: -1
|
||||
nms:
|
||||
name: MultiClassNMS
|
||||
nms_top_k: 1000
|
||||
keep_top_k: 300
|
||||
score_threshold: 0.01
|
||||
nms_threshold: 0.7
|
||||
Reference in New Issue
Block a user