English | 简体中文
- Apply MBLABlock(Multi Branch Layer Aggregation Block) blocks in the network structure.
Advantage:
-
Adopt a unified network structure and configuration.
-
Better performance for Small model comparing to yolov6 3.0 release.
-
Better performance comparing to yolov6 3.0 base.
| Model | Size | mAPval 0.5:0.95 |
SpeedT4 trt fp16 b1 (fps) |
SpeedT4 trt fp16 b32 (fps) |
Params (M) |
FLOPs (G) |
|---|---|---|---|---|---|---|
| YOLOv6-S-mbla | 640 | 47.0distill | 300 | 424 | 11.6 | 29.8 |
| YOLOv6-M-mbla | 640 | 50.3distill | 168 | 216 | 26.1 | 66.7 |
| YOLOv6-L-mbla | 640 | 52.0distill | 129 | 154 | 46.3 | 118.2 |
| YOLOv6-X-base | 640 | 53.5distill | 78 | 94 | 78.8 | 199.0 |
- Speed is tested with TensorRT 8.4.2.4 on T4.
- The processes of model training, evaluation, and inference are the same as the original ones. For details, please refer to this README.