Skip to content

Commit 5fc01a4

Browse files
hhaAndroidZwwWayne
authored andcommitted
Release YOLOX model (#6698)
* Release YOLOX model * update * update * update * update
1 parent 1eb94a5 commit 5fc01a4

File tree

3 files changed

+43
-1
lines changed

3 files changed

+43
-1
lines changed

configs/yolox/README.md

+6-1
Original file line numberDiff line numberDiff line change
@@ -31,9 +31,14 @@ In this report, we present some experienced improvements to YOLO series, forming
3131

3232
| Backbone | size | Mem (GB) | box AP | Config | Download |
3333
|:---------:|:-------:|:-------:|:-------:|:--------:|:------:|
34+
| YOLOX-tiny | 416 | 3.5 | 32.0 | [config](https://github.com/open-mmlab/mmdetection/tree/master/configs/yolox/yolox_tiny_8x8_300e_coco.py) |[model](https://download.openmmlab.com/mmdetection/v2.0/yolox/yolox_tiny_8x8_300e_coco/yolox_tiny_8x8_300e_coco_20211124_171234-b4047906.pth) | [log](https://download.openmmlab.com/mmdetection/v2.0/yolox/yolox_tiny_8x8_300e_coco/yolox_tiny_8x8_300e_coco_20211124_171234.log.json) |
3435
| YOLOX-s | 640 | 7.6 | 40.5 | [config](https://github.com/open-mmlab/mmdetection/tree/master/configs/yolox/yolox_s_8x8_300e_coco.py) |[model](https://download.openmmlab.com/mmdetection/v2.0/yolox/yolox_s_8x8_300e_coco/yolox_s_8x8_300e_coco_20211121_095711-4592a793.pth) | [log](https://download.openmmlab.com/mmdetection/v2.0/yolox/yolox_s_8x8_300e_coco/yolox_s_8x8_300e_coco_20211121_095711.log.json) |
36+
| YOLOX-l | 640 | 19.9 | 49.4 | [config](https://github.com/open-mmlab/mmdetection/tree/master/configs/yolox/yolox_l_8x8_300e_coco.py) |[model](https://download.openmmlab.com/mmdetection/v2.0/yolox/yolox_l_8x8_300e_coco/yolox_l_8x8_300e_coco_20211126_140236-d3bd2b23.pth) | [log](https://download.openmmlab.com/mmdetection/v2.0/yolox/yolox_l_8x8_300e_coco/yolox_l_8x8_300e_coco_20211126_140236.log.json) |
37+
| YOLOX-x | 640 | 28.1 | 50.9 | [config](https://github.com/open-mmlab/mmdetection/tree/master/configs/yolox/yolox_x_8x8_300e_coco.py) |[model](https://download.openmmlab.com/mmdetection/v2.0/yolox/yolox_x_8x8_300e_coco/yolox_x_8x8_300e_coco_20211126_140254-1ef88d67.pth) | [log](https://download.openmmlab.com/mmdetection/v2.0/yolox/yolox_x_8x8_300e_coco/yolox_x_8x8_300e_coco_20211126_140254.log.json) |
3538

3639

3740
**Note**:
3841

39-
1. The test score threshold is 0.001.
42+
1. The test score threshold is 0.001, and the box AP indicates the best AP.
43+
2. Due to the need for pre-training weights, we cannot reproduce the performance of the `yolox-nano` model. Please refer to https://github.com/Megvii-BaseDetection/YOLOX/issues/674 for more information.
44+
3. We also trained the model by the official release of YOLOX based on [Megvii-BaseDetection/YOLOX#735](https://github.com/Megvii-BaseDetection/YOLOX/issues/735) with commit ID [38c633](https://github.com/Megvii-BaseDetection/YOLOX/tree/38c633bf176462ee42b110c70e4ffe17b5753208). We found that the best AP of `YOLOX-tiny`, `YOLOX-s`, `YOLOX-l`, and `YOLOX-x` is 31.8, 40.3, 49.2, and 50.9, respectively. The performance is consistent with that of our re-implementation (see Table above) but still has a gap (0.3~0.8 AP) in comparison with the reported performance in their [README](https://github.com/Megvii-BaseDetection/YOLOX/blob/38c633bf176462ee42b110c70e4ffe17b5753208/README.md#benchmark).

configs/yolox/metafile.yml

+36
Original file line numberDiff line numberDiff line change
@@ -32,3 +32,39 @@ Models:
3232
Metrics:
3333
box AP: 40.5
3434
Weights: https://download.openmmlab.com/mmdetection/v2.0/yolox/yolox_s_8x8_300e_coco/yolox_s_8x8_300e_coco_20211121_095711-4592a793.pth
35+
- Name: yolox_l_8x8_300e_coco
36+
In Collection: YOLOX
37+
Config: configs/yolox/yolox_l_8x8_300e_coco.py
38+
Metadata:
39+
Training Memory (GB): 19.9
40+
Epochs: 300
41+
Results:
42+
- Task: Object Detection
43+
Dataset: COCO
44+
Metrics:
45+
box AP: 49.4
46+
Weights: https://download.openmmlab.com/mmdetection/v2.0/yolox/yolox_l_8x8_300e_coco/yolox_l_8x8_300e_coco_20211126_140236-d3bd2b23.pth
47+
- Name: yolox_x_8x8_300e_coco
48+
In Collection: YOLOX
49+
Config: configs/yolox/yolox_x_8x8_300e_coco.py
50+
Metadata:
51+
Training Memory (GB): 28.1
52+
Epochs: 300
53+
Results:
54+
- Task: Object Detection
55+
Dataset: COCO
56+
Metrics:
57+
box AP: 50.9
58+
Weights: https://download.openmmlab.com/mmdetection/v2.0/yolox/yolox_x_8x8_300e_coco/yolox_x_8x8_300e_coco_20211126_140254-1ef88d67.pth
59+
- Name: yolox_tiny_8x8_300e_coco
60+
In Collection: YOLOX
61+
Config: configs/yolox/yolox_tiny_8x8_300e_coco.py
62+
Metadata:
63+
Training Memory (GB): 3.5
64+
Epochs: 300
65+
Results:
66+
- Task: Object Detection
67+
Dataset: COCO
68+
Metrics:
69+
box AP: 32.0
70+
Weights: https://download.openmmlab.com/mmdetection/v2.0/yolox/yolox_tiny_8x8_300e_coco/yolox_tiny_8x8_300e_coco_20211124_171234-b4047906.pth

mmdet/datasets/pipelines/transforms.py

+1
Original file line numberDiff line numberDiff line change
@@ -2201,6 +2201,7 @@ class MixUp:
22012201
"""MixUp data augmentation.
22022202
22032203
.. code:: text
2204+
22042205
mixup transform
22052206
+------------------------------+
22062207
| mixup image | |

0 commit comments

Comments
 (0)