We have optimized and designed MAF-YOLOv2 based on the latest YOLO framework. This model achieves exceptionally high parameter efficiency and has reached state-of-the-art performance among all YOLO variants. The paper will be released soon.
Previous version: MAF-YOLO | Paper
- 2024/11/11: Add MAF-YOLOv2
MS COCO
| Model | Test Size | #Params | FLOPs | APval | AP50val |
|---|---|---|---|---|---|
| MAF-YOLOv2-Lite-N | 640 | 1.4M | 4.7G | 38.5% | 53.7% |
| MAF-YOLOv2-N | 640 | 2.2M | 7.2G | 42.3% | 58.5% |
| MAF-YOLOv2-S | 640 | 7.1M | 25.3G | 48.9% | 65.9% |
| MAF-YOLOv2-M | 640 | 15.3M | 65.2G | 52.7% | 69.5% |
Segmentation (COCO)
| Model | Test Size | #Params | FLOPs | APbbox | APmask |
|---|---|---|---|---|---|
| MAF-YOLOv2-N-Seg | 640 | 2.4M | 14.8G | 42.5% | 35.0% |
| MAF-YOLOv2-S-Seg | 640 | 7.8M | 40.4G | 48.8% | 39.7% |
conda create -n mafyolov2 python==3.9
conda activate mafyolov2
pip install torch==2.1.0 torchvision==0.16.0 torchaudio==2.1.0 --index-url https://download.pytorch.org/whl/cu121
pip install -r requirements.txt
pip install -e .
Single GPU training
# train.py
from ultralytics import YOLOv10
if __name__ == '__main__':
model = YOLOv10('MAF-YOLOv2-n.yaml')
model.train(data='coco.yaml', batch=16, device=0)If our code or model is helpful to your work, please cite our paper. We would be very grateful!
@InProceedings{10.1007/978-981-97-8858-3_34,
author="Yang, Zhiqiang
and Guan, Qiu
and Zhao, Keer
and Yang, Jianmin
and Xu, Xinli
and Long, Haixia
and Tang, Ying",
editor="Lin, Zhouchen
and Cheng, Ming-Ming
and He, Ran
and Ubul, Kurban
and Silamu, Wushouer
and Zha, Hongbin
and Zhou, Jie
and Liu, Cheng-Lin",
title="Multi-branch Auxiliary Fusion YOLO with Re-parameterization Heterogeneous Convolutional for Accurate Object Detection",
booktitle="Pattern Recognition and Computer Vision",
year="2025",
publisher="Springer Nature Singapore",
address="Singapore",
pages="492--505",
abstract="Due to the effective performance of multi-scale feature fusion, Path Aggregation FPN (PAFPN) is widely employed in YOLO detectors. However, it cannot efficiently and adaptively integrate high-level semantic information with low-level spatial information simultaneously. We propose a new model named MAF-YOLO in this paper, which is a novel object detection framework with a versatile neck named Multi-Branch Auxiliary FPN (MAFPN). Within MAFPN, the Superficial Assisted Fusion (SAF) module is designed to combine the output of the backbone with the neck, preserving an optimal level of shallow information to facilitate subsequent learning. Meanwhile, the Advanced Assisted Fusion (AAF) module deeply embedded within the neck conveys a more diverse range of gradient information to the output layer. Furthermore, our proposed Re-parameterized Heterogeneous Efficient Layer Aggregation Network (RepHELAN) module ensures that both the overall model architecture and convolutional design embrace the utilization of heterogeneous large convolution kernels. Therefore, this guarantees the preservation of information related to small targets while simultaneously achieving the multi-scale receptive field. Finally, taking the nano version of MAF-YOLO for example, it can achieve 42.4{\%} AP on COCO with only 3.76M learnable parameters and 10.51G FLOPs, and approximately outperforms YOLOv8n by about 5.1{\%}. The source code of this work is available at: https://github.com/yang-0201/MAF-YOLO.",
isbn="978-981-97-8858-3"
}or
@article{yang2024multi,
title={Multi-Branch Auxiliary Fusion YOLO with Re-parameterization Heterogeneous Convolutional for accurate object detection},
author={Yang, Zhiqiang and Guan, Qiu and Zhao, Keer and Yang, Jianmin and Xu, Xinli and Long, Haixia and Tang, Ying},
journal={arXiv preprint arXiv:2407.04381},
year={2024}
}