Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
s920243400
PaddleDetection
提交
3ef7245d
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看板
未验证
提交
3ef7245d
编写于
11月 03, 2021
作者:
G
Guanghua Yu
提交者:
GitHub
11月 03, 2021
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
add PicoDet voc config (#4443)
上级
f77456bf
变更
2
隐藏空白更改
内联
并排
Showing
2 changed file
with
93 addition
and
3 deletion
+93
-3
configs/picodet/README.md
configs/picodet/README.md
+56
-3
configs/picodet/picodet_s_320_voc.yml
configs/picodet/picodet_s_320_voc.yml
+37
-0
未找到文件。
configs/picodet/README.md
浏览文件 @
3ef7245d
...
...
@@ -20,9 +20,6 @@ We developed a series of lightweight models, named `PP-PicoDet`. Because of the
-
[ ] Pretrained models for more scenarios.
-
[ ] More features in need.
## Requirements
-
PaddlePaddle >= 2.1.2
## Benchmark
| Model | Input size | mAP
<sup>
val
<br>
0.5:0.95 | mAP
<sup>
val
<br>
0.5 | Params
<br><sup>
(M) | FLOPS
<br><sup>
(G) | Latency
<sup><small>
[
NCNN
](
#latency
)
</small><sup><br><sup>
(ms) | Latency
<sup><small>
[
Lite
](
#latency
)
</small><sup><br><sup>
(ms) | download | config |
...
...
@@ -69,6 +66,62 @@ We developed a series of lightweight models, named `PP-PicoDet`. Because of the
| YOLOv5s | 640
*
640 | 37.2 | 56.0 | 7.2 | 16.5 | 78.05 |
## Quick Start
<details
open
>
<summary>
Requirements:
</summary>
-
PaddlePaddle >= 2.1.2
</details>
<details>
<summary>
Installation
</summary>
-
[
Installation guide
](
https://github.com/PaddlePaddle/PaddleDetection/blob/release/2.3/docs/tutorials/INSTALL.md
)
-
[
Prepare dataset
](
https://github.com/PaddlePaddle/PaddleDetection/blob/release/2.3/docs/tutorials/PrepareDataSet_en.md
)
</details>
<details>
<summary>
Training and Evaluation
</summary>
-
Training model on single-GPU:
```
shell
# training on single-GPU
export
CUDA_VISIBLE_DEVICES
=
0
python tools/train.py
-c
configs/picodet/picodet_s_320_coco.yml
--eval
```
-
Training model on multi-GPU:
```
shell
# training on single-GPU
export
CUDA_VISIBLE_DEVICES
=
0,1,2,3,4,5,6,7
python
-m
paddle.distributed.launch
--gpus
0,1,2,3,4,5,6,7 tools/train.py
-c
configs/picodet/picodet_s_320_coco.yml
--eval
```
-
Evaluation:
```
shell
python tools/eval.py
-c
configs/picodet/picodet_s_320_coco.yml
\
-o
weights
=
https://paddledet.bj.bcebos.com/models/picodet_s_320_coco.pdparams
```
-
Infer:
```
shell
python tools/infer.py
-c
configs/picodet/picodet_s_320_coco.yml
\
-o
weights
=
https://paddledet.bj.bcebos.com/models/picodet_s_320_coco.pdparams
```
Detail also can refer to
[
Quick start guide
](
https://github.com/PaddlePaddle/PaddleDetection/blob/release/2.3/docs/tutorials/GETTING_STARTED.md
)
.
</details>
## Deployment
### Export and Convert Model
...
...
configs/picodet/picodet_s_320_voc.yml
0 → 100644
浏览文件 @
3ef7245d
_BASE_
:
[
'
../datasets/voc.yml'
,
'
../runtime.yml'
,
'
_base_/picodet_esnet.yml'
,
'
_base_/optimizer_300e.yml'
,
'
_base_/picodet_320_reader.yml'
,
]
pretrain_weights
:
https://paddledet.bj.bcebos.com/models/pretrained/ESNet_x0_75_pretrained.pdparams
weights
:
output/picodet_s_320_coco/model_final
find_unused_parameters
:
True
use_ema
:
true
cycle_epoch
:
40
snapshot_epoch
:
10
ESNet
:
scale
:
0.75
feature_maps
:
[
4
,
11
,
14
]
act
:
hard_swish
channel_ratio
:
[
0.875
,
0.5
,
0.5
,
0.5
,
0.625
,
0.5
,
0.625
,
0.5
,
0.5
,
0.5
,
0.5
,
0.5
,
0.5
]
CSPPAN
:
out_channels
:
96
PicoHead
:
conv_feat
:
name
:
PicoFeat
feat_in
:
96
feat_out
:
96
num_convs
:
2
num_fpn_stride
:
4
norm_type
:
bn
share_cls_reg
:
True
feat_in_chan
:
96
EvalReader
:
collate_batch
:
false
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录