mirrors/mmdeploy
1
0
Fork 0
mirror of https://github.com/open-mmlab/mmdeploy.git synced 2025-01-14 08:09:43 +08:00
Code Issues Projects Releases Wiki Activity
mmdeploy/README.md
hanrui1sensetime 4f530bc758
[Docs] Add docs of ncnn deployment (#33) 
* add docs

* Fix end-of-file error

* Fix end-of-file error

* add ncnn in README.md

* reply to AllentDan comments

* fix grammar error of docs
2021-08-10 17:04:58 +08:00

987 B
Raw Blame History

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