未验证 提交 61f1366f 编写于 作者: W wangguanzhong 提交者: GitHub

update yolov4 (#617)

上级 78c37c48
...@@ -10,9 +10,9 @@ ...@@ -10,9 +10,9 @@
[YOLO v4](https://arxiv.org/abs/2004.10934)的Paddle实现版本 [YOLO v4](https://arxiv.org/abs/2004.10934)的Paddle实现版本
目前PaddleDetection中转换了[darknet](https://github.com/AlexeyAB/darknet)中YOLO v4的权重,可以直接对图片进行预测,在[test-dev2019](http://cocodataset.org/#detection-2019)中精度为43.5%。另外,PaddleDetection支持VOC数据集上finetune,精度达到86.0% 目前转换了[darknet](https://github.com/AlexeyAB/darknet)中YOLO v4的权重,可以直接对图片进行预测,在[test-dev2019](http://cocodataset.org/#detection-2019)中精度为43.5%。另外,支持VOC数据集上finetune,精度达到85.5%
PaddleDetection支持YOLO v4的多个模块: 目前支持YOLO v4的多个模块:
- mish激活函数 - mish激活函数
- PAN模块 - PAN模块
...@@ -21,12 +21,12 @@ PaddleDetection支持YOLO v4的多个模块: ...@@ -21,12 +21,12 @@ PaddleDetection支持YOLO v4的多个模块:
- label_smooth - label_smooth
## 模型库 ## 模型库
下表中展示了PaddleDetection当前支持的网络结构。 下表中展示了当前支持的网络结构。
| | GPU个数 | 测试集 | 骨干网络 | 精度 | 模型下载 | 配置文件 | | | GPU个数 | 测试集 | 骨干网络 | 精度 | 模型下载 | 配置文件 |
|:------------------------:|:-------:|:------:|:--------------------------:|:------------------------:| :---------:| :-----: | |:------------------------:|:-------:|:------:|:--------------------------:|:------------------------:| :---------:| :-----: |
| YOLO v4 | - |test-dev2019 | CSPDarkNet53 | 43.5 |[下载链接](https://paddlemodels.bj.bcebos.com/object_detection/yolov4_cspdarknet.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/master/configs/yolo/yolov4_cspdarknet.yml) | | YOLO v4 | - |test-dev2019 | CSPDarkNet53 | 43.5 |[下载链接](https://paddlemodels.bj.bcebos.com/object_detection/yolov4_cspdarknet.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/master/configs/yolo/yolov4_cspdarknet.yml) |
| YOLO v4 VOC | 2 | VOC2007 | CSPDarkNet53 | - | [下载链接](https://paddlemodels.bj.bcebos.com/object_detection/yolov4_cspdarknet_voc.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/master/configs/yolo/yolov4_cspdarknet_voc.yml) | | YOLO v4 VOC | 2 | VOC2007 | CSPDarkNet53 | 85.5 | [下载链接](https://paddlemodels.bj.bcebos.com/object_detection/yolov4_cspdarknet_voc.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/master/configs/yolo/yolov4_cspdarknet_voc.yml) |
**注意:** **注意:**
......
architecture: YOLOv4 architecture: YOLOv4
use_gpu: true use_gpu: true
max_iters: 70000 max_iters: 140000
log_smooth_window: 20 log_smooth_window: 20
save_dir: output save_dir: output
snapshot_iter: 1000 snapshot_iter: 1000
...@@ -58,8 +58,8 @@ LearningRate: ...@@ -58,8 +58,8 @@ LearningRate:
- !PiecewiseDecay - !PiecewiseDecay
gamma: 0.1 gamma: 0.1
milestones: milestones:
- 56000 - 110000
- 63000 - 130000
- !LinearWarmup - !LinearWarmup
start_factor: 0. start_factor: 0.
steps: 1000 steps: 1000
...@@ -86,10 +86,6 @@ TrainReader: ...@@ -86,10 +86,6 @@ TrainReader:
sample_transforms: sample_transforms:
- !DecodeImage - !DecodeImage
to_rgb: True to_rgb: True
with_mixup: True
- !MixupImage
alpha: 1.5
beta: 1.5
- !ColorDistort {} - !ColorDistort {}
- !RandomExpand - !RandomExpand
fill_value: [123.675, 116.28, 103.53] fill_value: [123.675, 116.28, 103.53]
...@@ -123,7 +119,6 @@ TrainReader: ...@@ -123,7 +119,6 @@ TrainReader:
downsample_ratios: [8, 16, 32] downsample_ratios: [8, 16, 32]
batch_size: 4 batch_size: 4
shuffle: true shuffle: true
mixup_epoch: 28
drop_last: true drop_last: true
worker_num: 8 worker_num: 8
bufsize: 16 bufsize: 16
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册