MindYOLO

MindYOLO is MindSpore Lab's software toolbox that implements state-of-the-art YOLO series algorithms, support list and benchmark. It is written in Python and powered by the MindSpore AI framework.

The master branch supporting MindSpore 2.0/2.1.

What is New

• 2023/09/05

1. Add YOLOv8-X segment model.
2. Dataset pipeline reconstruction(current supports seg/detect tasks).
3. Add IoU custom operators example on GPU.
4. Add distribute eval function.
5. Add fast coco eval api.
6. Tutorials and Docs update(e.q. Write a new model, Train Process Tutorial, ...).

Benchmark and Model Zoo

See MODEL ZOO.

Supported Algorithms

• YOLOv8
• YOLOv7
• YOLOX
• YOLOv5
• YOLOv4
• YOLOv3

Installation

See INSTALLATION for details.

Getting Started

See GETTING STARTED for details.

Learn More about MindYOLO

To be supplemented.

Notes

⚠️ The current version is based on the static shape of GRAPH. The dynamic shape of the PYNATIVE will be supported later. Please look forward to it.

How to Contribute

We appreciate all contributions including issues and PRs to make MindYOLO better.

Please refer to CONTRIBUTING.md for the contributing guideline.

License

MindYOLO is released under the Apache License 2.0.

Acknowledgement

MindYOLO is an open source project that welcome any contribution and feedback. We wish that the toolbox and benchmark could support the growing research community, reimplement existing methods, and develop their own new real-time object detection methods by providing a flexible and standardized toolkit.

Citation

If you find this project useful in your research, please consider cite:

@misc{MindSpore Object Detection YOLO 2023,
    title={{MindSpore Object Detection YOLO}:MindSpore Object Detection YOLO Toolbox and Benchmark},
    author={MindSpore YOLO Contributors},
    howpublished = {\url{https://github.com/mindspore-lab/mindyolo}},
    year={2023}
}