Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
s920243400
PaddleDetection
提交
b0482b8b
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看板
未验证
提交
b0482b8b
编写于
6月 08, 2022
作者:
F
Feng Ni
提交者:
GitHub
6月 08, 2022
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
[cherry-pick] update yolox cfg and doc (#6153)
上级
3708813e
变更
2
隐藏空白更改
内联
并排
Showing
2 changed file
with
26 addition
and
11 deletion
+26
-11
configs/yolox/README.md
configs/yolox/README.md
+17
-11
configs/yolox/yoloxv2_tiny_300e_coco.yml
configs/yolox/yoloxv2_tiny_300e_coco.yml
+9
-0
未找到文件。
configs/yolox/README.md
浏览文件 @
b0482b8b
...
...
@@ -10,22 +10,28 @@
## 模型库
### YOLOX on COCO
| 网络网络 | 输入尺寸 | 图片数/GPU | 学习率策略 | 模型推理耗时(ms) | Box AP | 下载链接 | 配置文件 |
| :------------- | :------- | :-------: | :------: | :---------: | :-----: | :-------------: | :-----: |
| YOLOX-nano | 416 | 8 | 300e | 2.3 | 26.1 |
[
下载链接
](
https://paddledet.bj.bcebos.com/models/yolox_nano_300e_coco.pdparams
)
|
[
配置文件
](
./yolox_nano_300e_coco.yml
)
|
| YOLOX-tiny | 416 | 8 | 300e | 2.8 | 32.9 |
[
下载链接
](
https://paddledet.bj.bcebos.com/models/yolox_tiny_300e_coco.pdparams
)
|
[
配置文件
](
./yolox_tiny_300e_coco.yml
)
|
| YOLOX-s | 640 | 8 | 300e | 3.0 | 40.4 |
[
下载链接
](
https://paddledet.bj.bcebos.com/models/yolox_s_300e_coco.pdparams
)
|
[
配置文件
](
./yolox_s_300e_coco.yml
)
|
| YOLOX-m | 640 | 8 | 300e | 5.8 | 46.9 |
[
下载链接
](
https://paddledet.bj.bcebos.com/models/yolox_m_300e_coco.pdparams
)
|
[
配置文件
](
./yolox_m_300e_coco.yml
)
|
| YOLOX-l | 640 | 8 | 300e | 9.3 | 50.1 |
[
下载链接
](
https://paddledet.bj.bcebos.com/models/yolox_l_300e_coco.pdparams
)
|
[
配置文件
](
./yolox_l_300e_coco.yml
)
|
| YOLOX-x | 640 | 8 | 300e | 16.6 | 51.8 |
[
下载链接
](
https://paddledet.bj.bcebos.com/models/yolox_x_300e_coco.pdparams
)
|
[
配置文件
](
./yolox_x_300e_coco.yml
)
|
| 网络网络 | 输入尺寸 | 图片数/GPU | 学习率策略 | 模型推理耗时(ms) | mAP
<sup>
val
<br>
0.5:0.95 | mAP
<sup>
val
<br>
0.5 | 下载链接 | 配置文件 |
| :------------- | :------- | :-------: | :------: | :---------: | :-----------: | :-----------: |:-------------: | :-----: |
| YOLOX-nano | 416 | 8 | 300e | 2.3 | 26.1 | 42.0 |
[
下载链接
](
https://paddledet.bj.bcebos.com/models/yolox_nano_300e_coco.pdparams
)
|
[
配置文件
](
./yolox_nano_300e_coco.yml
)
|
| YOLOX-tiny | 416 | 8 | 300e | 2.8 | 32.9 | 50.4 |
[
下载链接
](
https://paddledet.bj.bcebos.com/models/yolox_tiny_300e_coco.pdparams
)
|
[
配置文件
](
./yolox_tiny_300e_coco.yml
)
|
| YOLOX-s | 640 | 8 | 300e | 3.0 | 40.4 | 59.6 |
[
下载链接
](
https://paddledet.bj.bcebos.com/models/yolox_s_300e_coco.pdparams
)
|
[
配置文件
](
./yolox_s_300e_coco.yml
)
|
| YOLOX-m | 640 | 8 | 300e | 5.8 | 46.9 | 65.7|
[
下载链接
](
https://paddledet.bj.bcebos.com/models/yolox_m_300e_coco.pdparams
)
|
[
配置文件
](
./yolox_m_300e_coco.yml
)
|
| YOLOX-l | 640 | 8 | 300e | 9.3 | 50.1 | 68.8 |
[
下载链接
](
https://paddledet.bj.bcebos.com/models/yolox_l_300e_coco.pdparams
)
|
[
配置文件
](
./yolox_l_300e_coco.yml
)
|
| YOLOX-x | 640 | 8 | 300e | 16.6 | 51.8 | 70.6 |
[
下载链接
](
https://paddledet.bj.bcebos.com/models/yolox_x_300e_coco.pdparams
)
|
[
配置文件
](
./yolox_x_300e_coco.yml
)
|
| 网络网络 | 输入尺寸 | 图片数/GPU | 学习率策略 | 模型推理耗时(ms) | mAP
<sup>
val
<br>
0.5:0.95 | mAP
<sup>
val
<br>
0.5 | 下载链接 | 配置文件 |
| :------------- | :------- | :-------: | :------: | :---------: | :-----------: | :-----------: |:-------------: | :-----: |
| YOLOXv2-tiny | 416 | 8 | 300e | 1.9 | 32.4 | 50.2 |
[
下载链接
](
https://paddledet.bj.bcebos.com/models/yoloxv2_tiny_300e_coco.pdparams
)
|
[
配置文件
](
./yoloxv2_tiny_300e_coco.yml
)
|
**注意:**
-
YOLOX模型训练使用COCO train2017作为训练集,
Box AP为在COCO val2017上的
`mAP(IoU=0.5:0.95)`
结果
;
-
YOLOX模型训练使用COCO train2017作为训练集,
YOLOXv2表示使用与YOLOv5 releases v6.0之后版本相同的主干网络
;
-
YOLOX模型训练过程中默认使用8 GPUs进行混合精度训练,默认每卡batch_size为8,默认lr为0.01为8卡总batch_size=64的设置,如果
**GPU卡数**
或者每卡
**batch size**
发生了改变,你需要按照公式
**lr<sub>new</sub> = lr<sub>default</sub> * (batch_size<sub>new</sub> * GPU_number<sub>new</sub>) / (batch_size<sub>default</sub> * GPU_number<sub>default</sub>)**
调整学习率;
-
为保持高mAP的同时提高推理速度,可以将
[
yolox_cspdarknet.yml
](
_base_/yolox_cspdarknet.yml
)
中的
`nms_top_k`
修改为
`1000`
,将
`keep_top_k`
修改为
`100`
,将
`score_threshold`
修改为
`0.01`
,mAP会下降约0.1~0.2%;
-
为快速的demo演示效果,可以将
[
yolox_cspdarknet.yml
](
_base_/yolox_cspdarknet.yml
)
中的
`score_threshold`
修改为
`0.25`
,将
`nms_threshold`
修改为
`0.45`
,但mAP会下降较多;
-
YOLOX模型推理速度测试采用单卡V100,batch size=1进行测试,使用
**CUDA 10.2**
,
**CUDNN 7.6.5**
,TensorRT推理速度测试使用
**TensorRT 6.0.1.8**
。
-
参考
[
速度测试
](
#速度测试
)
以复现YOLOX推理速度测试结果,速度为
tensorRT-FP16测速后的最快速度,不包含数据预处理和模型输出后处理(NMS)
的耗时。
-
参考
[
速度测试
](
#速度测试
)
以复现YOLOX推理速度测试结果,速度为
**tensorRT-FP16**
测速后的最快速度,
**不包含数据预处理和模型输出后处理(NMS)**
的耗时。
-
如果你设置了
`--run_benchmark=True`
, 你首先需要安装以下依赖
`pip install pynvml psutil GPUtil`
。
## 使用教程
...
...
@@ -154,7 +160,7 @@ python deploy/python/infer.py --model_dir=output_inference/yolox_s_300e_coco --i
```
**注意:**
-
导出模型时指定
`-o exclude_nms=True`
仅作为测速时用,这样导出的模型其推理部署预测的结果不是最终检出框的结果。
-
[
模型库
](
#模型库
)
中的速度测试结果为
tensorRT-FP16测速后的最快速度,为不包含数据预处理和模型输出后处理(NMS)
的耗时。
-
[
模型库
](
#模型库
)
中的速度测试结果为
**tensorRT-FP16**
测速后的最快速度,为
**不包含数据预处理和模型输出后处理(NMS)**
的耗时。
## Citations
...
...
configs/yolox/yoloxv2_tiny_300e_coco.yml
0 → 100644
浏览文件 @
b0482b8b
_BASE_
:
[
'
yolox_tiny_300e_coco.yml'
]
weights
:
output/yoloxv2_tiny_300e_coco/model_final
CSPDarkNet
:
arch
:
"
P5"
# using the same backbone of YOLOv5 releases v6.0 and later version
return_idx
:
[
2
,
3
,
4
]
depthwise
:
False
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录