OpenMMLab website HOT      OpenMMLab platform TRY IT OUT
 
[![PyPI](https://img.shields.io/pypi/v/mmyolo)](https://pypi.org/project/mmyolo) [![docs](https://img.shields.io/badge/docs-latest-blue)](https://mmyolo.readthedocs.io/en/latest/) [![badge](https://github.com/open-mmlab/mmyolo/workflows/build/badge.svg)](https://github.com/open-mmlab/mmyolo/actions) [![codecov](https://codecov.io/gh/open-mmlab/mmyolo/branch/master/graph/badge.svg)](https://codecov.io/gh/open-mmlab/mmyolo) [![license](https://img.shields.io/github/license/open-mmlab/mmyolo.svg)](https://github.com/open-mmlab/mmyolo/blob/master/LICENSE) [![open issues](https://isitmaintained.com/badge/open/open-mmlab/mmyolo.svg)](https://github.com/open-mmlab/mmyolo/issues) [![issue resolution](https://isitmaintained.com/badge/resolution/open-mmlab/mmyolo.svg)](https://github.com/open-mmlab/mmyolo/issues) [📘Documentation](https://mmyolo.readthedocs.io/en/latest/) | [🛠️Installation](https://mmyolo.readthedocs.io/en/latest/install.html) | [👀Model Zoo](https://mmyolo.readthedocs.io/en/latest/model_zoo.html) | [🆕Update News](https://mmyolo.readthedocs.io/en/latest/notes/changelog.html) | [🤔Reporting Issues](https://github.com/open-mmlab/mmyolo/issues/new/choose)
English | [简体中文](README_zh-CN.md)
## Introduction MMYOLO is an open source toolbox for YOLO series algorithms based on PyTorch. It is a part of the [OpenMMLab](https://openmmlab.com/) project. The master branch works with **PyTorch 1.6+**.
Major features - **Fair and convenient algorithm evaluation** MMYOLO unifies the modules of various YOLO algorithms and provides a unified benchmark process. Users can compare and analyze in a fair and convenient way. - **Detailed introductory and advanced documentation** MMYOLO provides a series of documents from getting started, to model deployment, advanced guidelines, and algorithm analysis, making it easy for different users to get started and make extensions quickly. - **Modular Design** MMYOLO decompose the framework into different components and users can easily construct a customized model by combining different modules and training and testing strategies. BaseModule The picture is provided by RangeKing@GitHub, thank you very much!
## What's New **v0.1.0** was released on 21/9/2022: - Unified component interfaces based on [OpenMMLab 2.0](https://github.com/open-mmlab) and [MMDetection 3.0](https://github.com/open-mmlab/mmdetection/tree/3.x) - Support for YOLOv5/YOLOX training and deployment, support for YOLOv6 inference and deployment - Refactored YOLOX for MMDetection to provide faster training and inference - Detailed introductory and advanced tutorials are provided, see the [English tutorial](https://mmyolo.readthedocs.io/en/latest) For release history and update details, please refer to [changelog](https://mmyolo.readthedocs.io/en/latest/notes/changelog.html). ## Installation MMYOLO relies on PyTorch, MMCV, MMEngine, and MMDetection. Below are quick steps for installation. Please refer to the [Install Guide](docs/en/get_started.md) for more detailed instructions. ```shell conda create -n open-mmlab python=3.8 pytorch==1.10.1 torchvision==0.11.2 cudatoolkit=11.3 -c pytorch -y conda activate open-mmlab pip install openmim mim install mmengine mim install "mmcv>=2.0.0rc1" mim install "mmdet>=3.0.0rc0" # for albumentations pip install -r requirements/albu.txt git clone https://github.com/open-mmlab/mmyolo.git cd mmyolo mim install -e . ``` ## Tutorial MMYOLO is based on the MMDetection and uses the same code organization and design approach. To get better use of this, please read [MMDetection Overview](https://mmdetection.readthedocs.io/en/latest/get_started.html) for the first understanding of MMDetection. MMYOLO usage is almost identical to MMDetection and all tutorials are straightforward to use, you can also learn about [MMDetection User Guide and Advanced Guide](https://mmdetection.readthedocs.io/en/3.x/). For different sections than MMDetection, we have also prepared user guides and advanced guides, please read our [documentation](https://mmyolo.readthedocs.io/zenh_CN/latest/). - User Guides - [Train & Test](https://mmyolo.readthedocs.io/en/latest/user_guides/index.html#train-test) - [Learn about Configs with YOLOv5](docs/en/user_guides/config.md) - [From getting started to deployment](https://mmyolo.readthedocs.io/zh_CN/latest/user_guides/index.html#get-started-to-deployment) - [From getting started to deployment with YOLOv5](docs/en/user_guides/yolov5_tutorial.md) - [Useful Tools](https://mmdetection.readthedocs.io/en/latest/user_guides/index.html#useful-tools) - [Visualization](docs/en/user_guides/visualization.md) - [Useful Tools](docs/en/user_guides/useful_tools.md) - Algorithm description - [Essential Basics](https://mmyolo.readthedocs.io/zh_CN/latest/algorithm_descriptions/index.html#essential-basics) - [Model design-related instructions](docs/en/algorithm_descriptions/model_design.md) - [Algorithm principles and implementation](https://mmyolo.readthedocs.io/zh_CN/latest/algorithm_descriptions/index.html#algorithm-principles-and-implementation) - [Algorithm principles and implementation with YOLOv5](docs/en/algorithm_descriptions/yolov5_description.md) - Advanced Guides - [How to](https://mmyolo.readthedocs.io/en/3.x/advanced_guides/index.html#how-to) ## Overview of Benchmark and Model Zoo Results and models are available in the [model zoo](docs/en/model_zoo.md).
Supported Algorithms - [x] [YOLOv5](configs/yolov5) - [x] [YOLOX](configs/yolox) - [ ] [YOLOv6](configs/yolov6)(Inference only)
Module Components
Backbones Necks Loss Common
  • YOLOv5CSPDarknet
  • YOLOXCSPDarknet
  • EfficientRep
  • YOLOv5PAFPN
  • YOLOv6RepPAFPN
  • YOLOXPAFPN
  • IoULoss
## FAQ Please refer to the [FAQ](docs/en/notes/faq.md) for frequently asked questions. ## Contributing We appreciate all contributions to improving MMYOLO. Ongoing projects can be found in our [GitHub Projects](https://github.com/open-mmlab/mmyolo/projects). Welcome community users to participate in these projects. Please refer to [CONTRIBUTING.md](.github/CONTRIBUTING.md) for the contributing guideline. ## Acknowledgement MMYOLO is an open source project that is contributed by researchers and engineers from various colleges and companies. We appreciate all the contributors who implement their methods or add new features, as well as users who give valuable feedback. We wish that the toolbox and benchmark could serve the growing research community by providing a flexible toolkit to reimplement existing methods and develop their own new detectors. ## Citation If you find this project useful in your research, please consider cite: ```latex @misc{mmyolo2022, title={{MMYOLO: OpenMMLab YOLO} series toolbox and benchmark}, author={MMYOLO Contributors}, howpublished = {\url{https://github.com/open-mmlab/mmyolo}}, year={2022} } ``` ## License This project is released under the [GPL 3.0 license](LICENSE). ## Projects in OpenMMLab - [MMEngine](https://github.com/open-mmlab/mmengine): OpenMMLab foundational library for training deep learning models. - [MMCV](https://github.com/open-mmlab/mmcv): OpenMMLab foundational library for computer vision. - [MIM](https://github.com/open-mmlab/mim): MIM installs OpenMMLab packages. - [MMClassification](https://github.com/open-mmlab/mmclassification): OpenMMLab image classification toolbox and benchmark. - [MMDetection](https://github.com/open-mmlab/mmdetection): OpenMMLab detection toolbox and benchmark. - [MMDetection3D](https://github.com/open-mmlab/mmdetection3d): OpenMMLab's next-generation platform for general 3D object detection. - [MMRotate](https://github.com/open-mmlab/mmrotate): OpenMMLab rotated object detection toolbox and benchmark. - [MMYOLO](https://github.com/open-mmlab/mmyolo): OpenMMLab YOLO series toolbox and benchmark. - [MMSegmentation](https://github.com/open-mmlab/mmsegmentation): OpenMMLab semantic segmentation toolbox and benchmark. - [MMOCR](https://github.com/open-mmlab/mmocr): OpenMMLab text detection, recognition, and understanding toolbox. - [MMPose](https://github.com/open-mmlab/mmpose): OpenMMLab pose estimation toolbox and benchmark. - [MMHuman3D](https://github.com/open-mmlab/mmhuman3d): OpenMMLab 3D human parametric model toolbox and benchmark. - [MMSelfSup](https://github.com/open-mmlab/mmselfsup): OpenMMLab self-supervised learning toolbox and benchmark. - [MMRazor](https://github.com/open-mmlab/mmrazor): OpenMMLab model compression toolbox and benchmark. - [MMFewShot](https://github.com/open-mmlab/mmfewshot): OpenMMLab fewshot learning toolbox and benchmark. - [MMAction2](https://github.com/open-mmlab/mmaction2): OpenMMLab's next-generation action understanding toolbox and benchmark. - [MMTracking](https://github.com/open-mmlab/mmtracking): OpenMMLab video perception toolbox and benchmark. - [MMFlow](https://github.com/open-mmlab/mmflow): OpenMMLab optical flow toolbox and benchmark. - [MMEditing](https://github.com/open-mmlab/mmediting): OpenMMLab image and video editing toolbox. - [MMGeneration](https://github.com/open-mmlab/mmgeneration): OpenMMLab image and video generative models toolbox. - [MMDeploy](https://github.com/open-mmlab/mmdeploy): OpenMMLab model deployment framework.