mirrors
/
mmdeploy
mirror of https://github.com/open-mmlab/mmdeploy.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
OpenMMLab Model Deployment Framework
47 Commits
16 Branches
23 Tags
187 MiB
Python 46.6%
C++ 41.3%
Cuda 4.4%
CMake 2%
C# 1.9%
Other 3.8%
 
 
 
 
 
 
Go to file
hanrui1sensetime f8b62ae4d0
[Feature] Support NCNN inference tests (#34) 
* add inference test

* fix for review comments

* add inference test results table

* fix grammer errors

* fix typo
 		2021-08-12 17:38:38 +08:00
.github/workflows remove unused flag, fix lint 2021-06-29 10:23:52 +08:00
backend_ops [Feature] Merge NCNN deployment to grimoire based on mmcls - revert [#25](https://github.com/grimoire/deploy_prototype/pull/25) (#30) 2021-08-05 14:06:47 +08:00
configs [Feature]: Support FCN,DeeepLabV3, DeepLabV3Plus in mmseg with ONNXRuntime and TensorRT (#31) 2021-08-12 16:44:16 +08:00
docker add docker (#15) 2021-07-13 08:45:42 +00:00
docs [Feature] Support NCNN inference tests (#34) 2021-08-12 17:38:38 +08:00
mmdeploy [Feature] Support NCNN inference tests (#34) 2021-08-12 17:38:38 +08:00
tests [Feature] Merge NCNN deployment to grimoire based on mmcls - revert [#25](https://github.com/grimoire/deploy_prototype/pull/25) (#30) 2021-08-05 14:06:47 +08:00
third_party [Feature] Add TensorRT batched NMS support (#3) 2021-06-25 19:31:16 +08:00
tools [Feature] Support NCNN inference tests (#34) 2021-08-12 17:38:38 +08:00
.gitignore add register 2021-06-11 13:26:05 +08:00
.gitmodules [Feature] Add TensorRT batched NMS support (#3) 2021-06-25 19:31:16 +08:00
.isort.cfg [Feature]: Support FCN,DeeepLabV3, DeepLabV3Plus in mmseg with ONNXRuntime and TensorRT (#31) 2021-08-12 16:44:16 +08:00
.pre-commit-config.yaml add register 2021-06-11 13:26:05 +08:00
CMakeLists.txt [Feature] Merge NCNN deployment to grimoire based on mmcls - revert [#25](https://github.com/grimoire/deploy_prototype/pull/25) (#30) 2021-08-05 14:06:47 +08:00
LICENSE
…
README.md [Docs] Add docs of ncnn deployment (#33) 2021-08-10 17:04:58 +08:00
setup.cfg add type annotation 2021-06-16 15:36:58 +08:00
setup.py add register 2021-06-11 13:26:05 +08:00

README.md

MMDeployment

Installation

  • Build backend ops

    • update submodule

      git submodule update --init

    • Build with onnxruntime support

      mkdir build
cd build
cmake -DBUILD_ONNXRUNTIME_OPS=ON -DONNXRUNTIME_DIR=${PATH_TO_ONNXRUNTIME} ..
make -j10

    • Build with tensorrt support

      mkdir build
cd build
cmake -DBUILD_TENSORRT_OPS=ON -DTENSORRT_DIR=${PATH_TO_TENSORRT} ..
make -j10

    • Build with ncnn support

      mkdir build
cd build
cmake -DBUILD_NCNN_OPS=ON -DNCNN_DIR=${PATH_TO_NCNN} ..
make -j10

    • Or you can add multiple flags to build multiple backend ops.

  • Setup project

    python setup.py develop

Usage

python ./tools/deploy.py \
    ${DEPLOY_CFG_PATH} \
    ${MODEL_CFG_PATH} \
    ${MODEL_CHECKPOINT_PATH} \
    ${INPUT_IMG} \
    --work-dir ${WORK_DIR} \
    --device ${DEVICE} \
    --log-level INFO