OpenMMLab Model Deployment Framework
 
 
 
 
 
 
Go to file
Johannes L 651eec239b
MMdeploy build & install script + prerequisites (#399)
* Initial commit of build script

* Fixed lint

* * Set prefix depending on install arch
* Auto detect installed version of cmake, gcc. Update if necessary
* Prompt to reinstall venv if exists
* Disable .tar packing
* Enable MMDEPLOY SDK Python
* Misc cleanup

* Fixed lint

* Renamed script

* * Set env variables without restarting script.
* OpenCV install optional.
* Fix numpy OPENBLAS_CORETYPE on numpy >1.19.4.
* Remove specific build files.
* Fix prebuild output .tar.gz.
* Comments added. exec bash when closing script

* * Use root dir as MMDeploy dir.
* Use pip3.
* Remove explicit MMDeploy checkout version.
* Use MMDEPLOY_DIR for examples path

* Make PyTorch, torchvision, mmcv install optional

* * Do not deactivate python venv.
* Make prebuild archive optional.
* Add venv* to gitignore

* * Add support for unattended, interactive install. * Cleanup  misc. comments.

* Check input arguments before continuing.

* Added additional log messages, comments added

* Added default option when pressing [ENTER]

* * Rearranged cmake, gcc build/install order.
* Mention default option when reinstalling venv
* sudo make install instead of cmake --install

* Moved script to tools/scripts

* Fixed relative path when generating WORKING_DIR
2022-05-25 17:13:24 +08:00
.circleci Add nms_rotated ort op (#312) 2022-04-18 17:14:47 +08:00
.github refactor(onnx2ncnn): add test case and simplify code (#436) 2022-05-16 10:36:25 +08:00
cmake mmrotate sdk module (#450) 2022-05-17 23:37:32 +08:00
configs [Enhancement] Support RotatedRetinaNet TensorRT (#422) 2022-05-24 10:34:22 +08:00
csrc [Enhancement] Support RotatedRetinaNet TensorRT (#422) 2022-05-24 10:34:22 +08:00
demo mmrotate sdk module (#450) 2022-05-17 23:37:32 +08:00
docker Add nms_rotated ort op (#312) 2022-04-18 17:14:47 +08:00
docs [Feature] Support two stage rotated detector in MMRotate (#473) 2022-05-25 13:41:16 +08:00
mmdeploy [Feature] Support two stage rotated detector in MMRotate (#473) 2022-05-25 13:41:16 +08:00
requirements [Refactor][tools] Add prebuild tools. (#347) 2022-05-23 17:15:09 +08:00
resources fix intro (#337) 2021-12-24 18:49:09 +08:00
tests [Feature] Support two stage rotated detector in MMRotate (#473) 2022-05-25 13:41:16 +08:00
third_party Merge sdk (#251) 2021-12-07 10:57:55 +08:00
tools MMdeploy build & install script + prerequisites (#399) 2022-05-25 17:13:24 +08:00
.clang-format Merge sdk (#251) 2021-12-07 10:57:55 +08:00
.gitignore MMdeploy build & install script + prerequisites (#399) 2022-05-25 17:13:24 +08:00
.gitmodules Merge sdk (#251) 2021-12-07 10:57:55 +08:00
.pre-commit-config.yaml Dev v0.4.0 (#301) 2022-04-01 18:14:23 +08:00
.pylintrc
.readthedocs.yml add gpu ci (#35) 2022-01-10 11:14:17 +08:00
CITATION.cff
CMakeLists.txt refactor(onnx2ncnn): add test case and simplify code (#436) 2022-05-16 10:36:25 +08:00
LICENSE
MANIFEST.in [Refactor][tools] Add prebuild tools. (#347) 2022-05-23 17:15:09 +08:00
README.md [enhancement] Reorganizing OpenMMLab projects in readme (#184) 2022-02-28 15:38:31 +08:00
README_zh-CN.md fix(zh_CN): doc link (#447) 2022-05-12 12:22:50 +08:00
requirements.txt
setup.cfg [Refactor][tools] Add prebuild tools. (#347) 2022-05-23 17:15:09 +08:00
setup.py [Refactor][tools] Add prebuild tools. (#347) 2022-05-23 17:15:09 +08:00

README.md

 
OpenMMLab website HOT      OpenMMLab platform TRY IT OUT
 

docs badge codecov license issue resolution open issues

Introduction

English | 简体中文

MMDeploy is an open-source deep learning model deployment toolset. It is a part of the OpenMMLab project.

Major features

  • Fully support OpenMMLab models

    We provide a unified model deployment toolbox for the codebases in OpenMMLab. The supported codebases are listed as below, and more will be added in the future

    • MMClassification
    • MMDetection
    • MMSegmentation
    • MMEditing
    • MMOCR
    • MMPose
  • Multiple inference backends are available

    Models can be exported and run in different backends. The following ones are supported, and more will be taken into consideration

    • ONNX Runtime
    • TensorRT
    • PPLNN
    • ncnn
    • OpenVINO
  • Efficient and highly scalable SDK Framework by C/C++

    All kinds of modules in SDK can be extensible, such as Transform for image processing, Net for Neural Network inference, Module for postprocessing and so on

License

This project is released under the Apache 2.0 license.

Installation

Please refer to build.md for installation.

Getting Started

Please see getting_started.md for the basic usage of MMDeploy. We also provide other tutorials for:

Please refer to FAQ for frequently asked questions.

Benchmark and model zoo

Results and supported model list are available in the benchmark and model list.

Contributing

We appreciate all contributions to improve MMDeploy. Please refer to CONTRIBUTING.md for the contributing guideline.

Acknowledgement

We would like to sincerely thank the following teams for their contributions to MMDeploy:

Citation

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

@misc{=mmdeploy,
    title={OpenMMLab's Model Deployment Toolbox.},
    author={MMDeploy Contributors},
    howpublished = {\url{https://github.com/open-mmlab/mmdeploy}},
    year={2021}
}

Projects in OpenMMLab

  • MMCV: OpenMMLab foundational library for computer vision.
  • MIM: MIM installs OpenMMLab packages.
  • MMClassification: OpenMMLab image classification toolbox and benchmark.
  • MMDetection: OpenMMLab detection toolbox and benchmark.
  • MMDetection3D: OpenMMLab's next-generation platform for general 3D object detection.
  • MMRotate: OpenMMLab rotated object detection toolbox and benchmark.
  • MMSegmentation: OpenMMLab semantic segmentation toolbox and benchmark.
  • MMOCR: OpenMMLab text detection, recognition, and understanding toolbox.
  • MMPose: OpenMMLab pose estimation toolbox and benchmark.
  • MMHuman3D: OpenMMLab 3D human parametric model toolbox and benchmark.
  • MMSelfSup: OpenMMLab self-supervised learning toolbox and benchmark.
  • MMRazor: OpenMMLab model compression toolbox and benchmark.
  • MMFewShot: OpenMMLab fewshot learning toolbox and benchmark.
  • MMAction2: OpenMMLab's next-generation action understanding toolbox and benchmark.
  • MMTracking: OpenMMLab video perception toolbox and benchmark.
  • MMFlow: OpenMMLab optical flow toolbox and benchmark.
  • MMEditing: OpenMMLab image and video editing toolbox.
  • MMGeneration: OpenMMLab image and video generative models toolbox.
  • MMDeploy: OpenMMLab model deployment framework.