19 lines
698 B
Bash
19 lines
698 B
Bash
#!/bin/bash
|
|
|
|
if [ $# != 2 ]; then
|
|
echo "wrong command. usage: bash converter.sh <codebase> <work dir>"
|
|
exit 1
|
|
fi
|
|
|
|
if [ "$1" == 'mmcls' ]; then
|
|
python3 -m pip install mmcls
|
|
git clone --recursive https://github.com/open-mmlab/mmclassification.git
|
|
wget https://download.openmmlab.com/mmclassification/v0/resnet/resnet18_8xb32_in1k_20210831-fbbb1da6.pth
|
|
python3 mmdeploy/tools/deploy.py \
|
|
mmdeploy/configs/mmcls/classification_onnxruntime_dynamic.py \
|
|
mmclassification/configs/resnet/resnet18_8xb32_in1k.py \
|
|
resnet18_8xb32_in1k_20210831-fbbb1da6.pth \
|
|
mmclassification/demo/demo.JPEG \
|
|
--work-dir "$2" --dump-info
|
|
fi
|