Skip to content

Commit

Permalink
Merge pull request meituan#744 from meituan/updae_base_model
Browse files Browse the repository at this point in the history 
update readme of base models
  • Loading branch information
@mtjhl
mtjhl authored Mar 2, 2023
2 parents 44a5034 + 2d7af0e commit 15d866d
Show file tree
Hide file tree
Showing 4 changed files with 18 additions and 17 deletions.
2 changes: 1 addition & 1 deletion README.md
View file
Original file line number Diff line number Diff line change
Expand Up @@ -25,7 +25,7 @@ Implementation of paper:


## What's New

- [2023.3.2] Updates [base models](configs/base/README.md) to version 0.3.
- [2023.01.06] Release P6 models and enhance the performance of P5 models. ⭐️ [Benchmark](#Benchmark)
- Renew the neck of the detector with a BiC module and SimCSPSPPF Block.
- Propose an anchor-aided training (AAT) strategy.
Expand Down
1 change: 1 addition & 0 deletions README_cn.md
View file
Original file line number Diff line number Diff line change
Expand Up @@ -16,6 +16,7 @@


## 更新日志
- [2023.3.2] 更新 [基础版模型](configs/base/READM_cn.md) 到 0.3 版本。
- [2023.01.06] 发布大分辨率 P6 模型以及对 P5 模型做了全面的升级 ⭐️ [模型指标](#模型指标)
- 添加 BiC 模块 和 SimCSPSPPF 模块以增强检测网络颈部的表征能力。
- 提出一个锚点辅助训练 (AAT) 策略。
Expand Down
16 changes: 8 additions & 8 deletions configs/base/README.md
View file
Original file line number Diff line number Diff line change
Expand Up @@ -17,12 +17,12 @@ Shortcoming:

### Performance

| Model | Size | mAP<sup>val<br/>0.5:0.95 | Speed<sup>T4<br/>trt fp16 b1 <br/>(fps) | Speed<sup>T4<br/>trt fp16 b32 <br/>(fps) | Params<br/><sup> (M) | FLOPs<br/><sup> (G) |
| :----------------------------------------------------------- | -------- | :----------------------- | -------------------------------------- | --------------------------------------- | -------------------- | ------------------- |
| [**YOLOv6-N-base**](https://github.com/meituan/YOLOv6/releases/download/0.2.1/yolov6n_base.pt) | 640 | 35.6<sup>400e | 832 | 1249 | 4.3 | 11.1 |
| [**YOLOv6-S-base**](https://github.com/meituan/YOLOv6/releases/download/0.2.1/yolov6s_base.pt) | 640 | 43.8<sup>400e | 373 | 531 | 11.5 | 27.6 |
| [**YOLOv6-M-base**](https://github.com/meituan/YOLOv6/releases/download/0.2.1/yolov6m_base.pt) | 640 | 48.8<sup>distill | 179 | 246 | 27.7 | 68.4 |
| [**YOLOv6-L-base**](https://github.com/meituan/YOLOv6/releases/download/0.2.1/yolov6l_base.pt) | 640 | 51.0<sup>distill | 115 | 153 | 58.5 | 144.0 |

- Speed is tested with TensorRT 7.2 on T4.
| Model | Size | mAP<sup>val<br/>0.5:0.95 | Speed<sup>T4<br/>TRT FP16 b1 <br/>(FPS) | Speed<sup>T4<br/>TRT FP16 b32 <br/>(FPS) | Speed<sup>T4<br/>TRT INT8 b1 <br/>(FPS) | Speed<sup>T4<br/>TRT INT8 b32 <br/>(FPS) | Params<br/><sup> (M) | FLOPs<br/><sup> (G) |
| :--------------------------------------------------------------------------------------------- | --- | ----------------- | ----- | ---- | ---- | ---- | ----- | ------ |
| [**YOLOv6-N-base**](https://github.com/meituan/YOLOv6/releases/download/0.2.1/yolov6n_base.pt) | 640 | 36.6<sup>distill | 727 | 1302 | 814 | 1805 | 4.65 | 11.46 |
| [**YOLOv6-S-base**](https://github.com/meituan/YOLOv6/releases/download/0.2.1/yolov6s_base.pt) | 640 | 45.3<sup>distill | 346 | 525 | 487 | 908 | 13.14 | 30.6 |
| [**YOLOv6-M-base**](https://github.com/meituan/YOLOv6/releases/download/0.2.1/yolov6m_base.pt) | 640 | 49.4<sup>distill | 179 | 245 | 284 | 439 | 28.33 | 72.30 |
| [**YOLOv6-L-base**](https://github.com/meituan/YOLOv6/releases/download/0.2.1/yolov6l_base.pt) | 640 | 51.1<sup>distill | 116 | 157 | 196 | 288 | 59.61 | 150.89 |

- Speed is tested with TensorRT 8.2.4.2 on T4.
- The processes of model training, evaluation, and inference are the same as the original ones. For details, please refer to [this README](https://github.com/meituan/YOLOv6#quick-start).
16 changes: 8 additions & 8 deletions configs/base/README_cn.md
View file
Original file line number Diff line number Diff line change
Expand Up @@ -16,12 +16,12 @@

### 模型指标

| 模型 | 输入尺寸 | mAP<sup>val<br/>0.5:0.95 | 速度<sup>T4<br/>trt fp16 b1 <br/>(fps) | 速度<sup>T4<br/>trt fp16 b32 <br/>(fps) | Params<br/><sup> (M) | FLOPs<br/><sup> (G) |
| :----------------------------------------------------------- | -------- | :----------------------- | -------------------------------------- | --------------------------------------- | -------------------- | ------------------- |
| [**YOLOv6-N-base**](https://github.com/meituan/YOLOv6/releases/download/0.2.1/yolov6n_base.pt) | 640 | 35.6<sup>400e | 832 | 1249 | 4.3 | 11.1 |
| [**YOLOv6-S-base**](https://github.com/meituan/YOLOv6/releases/download/0.2.1/yolov6s_base.pt) | 640 | 43.8<sup>400e | 373 | 531 | 11.5 | 27.6 |
| [**YOLOv6-M-base**](https://github.com/meituan/YOLOv6/releases/download/0.2.1/yolov6m_base.pt) | 640 | 48.8<sup>distill | 179 | 246 | 27.7 | 68.4 |
| [**YOLOv6-L-base**](https://github.com/meituan/YOLOv6/releases/download/0.2.1/yolov6l_base.pt) | 640 | 51.0<sup>distill | 115 | 153 | 58.5 | 144.0 |

- 速度是在 T4 上测试的，TensorRT 版本为 7.2；
| 模型 | 尺寸 | mAP<sup>val<br/>0.5:0.95 | 速度<sup>T4<br/>TRT FP16 b1 <br/>(FPS) | 速度<sup>T4<br/>TRT FP16 b32 <br/>(FPS) | 速度<sup>T4<br/>TRT INT8 b1 <br/>(FPS) | 速度<sup>T4<br/>TRT INT8 b32 <br/>(FPS) | Params<br/><sup> (M) | FLOPs<br/><sup> (G) |
| :--------------------------------------------------------------------------------------------- | --- | ----------------- | ----- | ---- | ---- | ---- | ----- | ------ |
| [**YOLOv6-N-base**](https://github.com/meituan/YOLOv6/releases/download/0.2.1/yolov6n_base.pt) | 640 | 36.6<sup>distill | 727 | 1302 | 814 | 1805 | 4.65 | 11.46 |
| [**YOLOv6-S-base**](https://github.com/meituan/YOLOv6/releases/download/0.2.1/yolov6s_base.pt) | 640 | 45.3<sup>distill | 346 | 525 | 487 | 908 | 13.14 | 30.6 |
| [**YOLOv6-M-base**](https://github.com/meituan/YOLOv6/releases/download/0.2.1/yolov6m_base.pt) | 640 | 49.4<sup>distill | 179 | 245 | 284 | 439 | 28.33 | 72.30 |
| [**YOLOv6-L-base**](https://github.com/meituan/YOLOv6/releases/download/0.2.1/yolov6l_base.pt) | 640 | 51.1<sup>distill | 116 | 157 | 196 | 288 | 59.61 | 150.89 |

- 速度是在 T4 上测试的，TensorRT 版本为 8.4.2.4；
- 模型训练、评估、推理流程与原来保持一致，具体可参考 [首页 README 文档](https://github.com/meituan/YOLOv6/blob/main/README_cn.md#%E5%BF%AB%E9%80%9F%E5%BC%80%E5%A7%8B)。

0 comments on commit 15d866d

Please sign in to comment.