2021-06-11 13:26:05 +08:00
|
|
|
# MMDeployment
|
|
|
|
|
2021-07-01 17:32:33 +08:00
|
|
|
## Installation
|
|
|
|
|
|
|
|
- Build backend ops
|
|
|
|
|
2021-07-10 16:16:11 +08:00
|
|
|
- update submodule
|
|
|
|
|
|
|
|
```bash
|
|
|
|
git submodule update --init
|
|
|
|
```
|
|
|
|
|
2021-07-01 17:32:33 +08:00
|
|
|
- Build with onnxruntime support
|
|
|
|
|
|
|
|
```bash
|
|
|
|
mkdir build
|
|
|
|
cd build
|
|
|
|
cmake -DBUILD_ONNXRUNTIME_OPS=ON -DONNXRUNTIME_DIR=${PATH_TO_ONNXRUNTIME} ..
|
|
|
|
make -j10
|
|
|
|
```
|
|
|
|
|
|
|
|
- Build with tensorrt support
|
|
|
|
|
|
|
|
```bash
|
|
|
|
mkdir build
|
|
|
|
cd build
|
|
|
|
cmake -DBUILD_TENSORRT_OPS=ON -DTENSORRT_DIR=${PATH_TO_TENSORRT} ..
|
|
|
|
make -j10
|
|
|
|
```
|
|
|
|
|
2021-08-10 17:04:58 +08:00
|
|
|
- Build with ncnn support
|
|
|
|
|
|
|
|
```bash
|
|
|
|
mkdir build
|
|
|
|
cd build
|
|
|
|
cmake -DBUILD_NCNN_OPS=ON -DNCNN_DIR=${PATH_TO_NCNN} ..
|
|
|
|
make -j10
|
|
|
|
```
|
|
|
|
|
2021-07-01 17:32:33 +08:00
|
|
|
- Or you can add multiple flags to build multiple backend ops.
|
|
|
|
|
|
|
|
- Setup project
|
|
|
|
|
|
|
|
```bash
|
|
|
|
python setup.py develop
|
|
|
|
```
|
|
|
|
|
|
|
|
## Usage
|
|
|
|
|
|
|
|
```bash
|
|
|
|
python ./tools/deploy.py \
|
|
|
|
${DEPLOY_CFG_PATH} \
|
|
|
|
${MODEL_CFG_PATH} \
|
|
|
|
${MODEL_CHECKPOINT_PATH} \
|
|
|
|
${INPUT_IMG} \
|
|
|
|
--work-dir ${WORK_DIR} \
|
|
|
|
--device ${DEVICE} \
|
|
|
|
--log-level INFO
|
|
|
|
```
|