mmcls -> mmpretrain (#2003)

* mmcls -> mmpretrain

* add constraints.txt

* fix lint

* fix lint

* remove constraints.txt

* fix windows ci

* modify requirements

* fix mdlink and mmpretrain version

* fix dead link

* modify codebase cmakelist

* fix rename

Author: huayuan4396

.github/scripts/linux/test_full_pipeline.sh

@@ -14,19 +14,19 @@ mkdir -p $work_dir $mmdeploy_dir/data
 
 model_cfg=$work_dir/resnet18_8xb32_in1k.py
 checkpoint=$work_dir/resnet18_8xb32_in1k_20210831-fbbb1da6.pth
-sdk_cfg=configs/mmcls/classification_sdk_dynamic.py
+sdk_cfg=configs/mmpretrain/classification_sdk_dynamic.py
 input_img=tests/data/tiger.jpeg
 
-python3 -m mim download mmcls --config resnet18_8xb32_in1k --dest $work_dir
+python3 -m mim download mmpretrain --config resnet18_8xb32_in1k --dest $work_dir
 
 if [ $backend == "ort" ]; then
-    deploy_cfg=configs/mmcls/classification_onnxruntime_dynamic.py
+    deploy_cfg=configs/mmpretrain/classification_onnxruntime_dynamic.py
     model=$work_dir/end2end.onnx
 elif [ $backend == "trt" ]; then
-    deploy_cfg=configs/mmcls/classification_tensorrt-fp16_dynamic-224x224-224x224.py
+    deploy_cfg=configs/mmpretrain/classification_tensorrt-fp16_dynamic-224x224-224x224.py
     model=$work_dir/end2end.engine
 elif [ $backend == "ncnn" ]; then
-    deploy_cfg=configs/mmcls/classification_ncnn_static.py
+    deploy_cfg=configs/mmpretrain/classification_ncnn_static.py
     model="$work_dir/end2end.param $work_dir/end2end.bin"
 else
   echo "Unsupported Backend=$backend"

.github/scripts/quantize_to_ncnn.py

@@ -9,17 +9,17 @@
     [
         'hrnet.onnx',
         'https://media.githubusercontent.com/media/tpoisonooo/mmdeploy-onnx2ncnn-testdata/main/hrnet.onnx',  # noqa: E501
-        '~/mmclassification/configs/hrnet/hrnet-w18_4xb32_in1k.py',
+        '~/mmpretrain/configs/hrnet/hrnet-w18_4xb32_in1k.py',
     ],
     [
         'resnet18.onnx',
         'https://media.githubusercontent.com/media/tpoisonooo/mmdeploy-onnx2ncnn-testdata/main/resnet18.onnx',  # noqa: E501
-        '~/mmclassification/configs/resnet/resnet18_8xb16_cifar10.py',
+        '~/mmpretrain/configs/resnet/resnet18_8xb16_cifar10.py',
     ],
     [
         'mobilenet-v2.onnx',
         'https://media.githubusercontent.com/media/tpoisonooo/mmdeploy-onnx2ncnn-testdata/main/mobilenet-v2.onnx',  # noqa: E501
-        '~/mmclassification/configs/mobilenet_v2/mobilenet-v2_8xb32_in1k.py',
+        '~/mmpretrain/configs/mobilenet_v2/mobilenet-v2_8xb32_in1k.py',
     ]
 ]
 
@@ -45,7 +45,7 @@ def main():
         model = conf[0]
         os.system('wget {}'.format(conf[1]))
         model_cfg = conf[2]
-        deploy_cfg = 'configs/mmcls/classification_ncnn-int8_static.py'
+        deploy_cfg = 'configs/mmpretrain/classification_ncnn-int8_static.py'
         quant_cmd = [
             'python', 'tools/onnx2ncnn_quant_table.py', '--onnx', model,
             '--deploy-cfg', deploy_cfg, '--model-cfg', model_cfg, '--out-onnx',

.github/scripts/test_onnx2ncnn.py

@@ -7,22 +7,22 @@
 # list of tuple: config, pretrained model, onnx filename
 CONFIGS = [
     (
-        'mmclassification/configs/vision_transformer/vit-base-p32_ft-64xb64_in1k-384.py',  # noqa: E501
+        'mmpretrain/configs/vision_transformer/vit-base-p32_ft-64xb64_in1k-384.py',  # noqa: E501
         'https://download.openmmlab.com/mmclassification/v0/vit/finetune/vit-base-p32_in21k-pre-3rdparty_ft-64xb64_in1k-384_20210928-9cea8599.pth',  # noqa: E501
         'vit.onnx'),
     (
-        'mmclassification/configs/resnet/resnet50_8xb32_in1k.py',
+        'mmpretrain/configs/resnet/resnet50_8xb32_in1k.py',
         'https://download.openmmlab.com/mmclassification/v0/resnet/resnet50_8xb32_in1k_20210831-ea4938fc.pth',  # noqa: E501
         'resnet50.onnx',
     ),
     (
-        'mmclassification/configs/resnet/resnet18_8xb32_in1k.py',
+        'mmpretrain/configs/resnet/resnet18_8xb32_in1k.py',
         'https://download.openmmlab.com/mmclassification/v0/resnet/resnet18_8xb32_in1k_20210831-fbbb1da6.pth',  # noqa: E501
         'resnet18.onnx',
         'https://media.githubusercontent.com/media/tpoisonooo/mmdeploy-onnx2ncnn-testdata/main/resnet18.onnx',  # noqa: E501
     ),
     (
-        'mmclassification/configs/mobilenet_v2/mobilenet-v2_8xb32_in1k.py',
+        'mmpretrain/configs/mobilenet_v2/mobilenet-v2_8xb32_in1k.py',
         'https://download.openmmlab.com/mmclassification/v0/mobilenet_v2/mobilenet_v2_batch256_imagenet_20200708-3b2dc3af.pth',  # noqa: E501
         'mobilenet-v2.onnx',
         'https://media.githubusercontent.com/media/tpoisonooo/mmdeploy-onnx2ncnn-testdata/main/mobilenet-v2.onnx',  # noqa: E501
@@ -36,7 +36,7 @@ def parse_args():
     parser.add_argument(
         '--run', type=bool, help='Execute mmdeploy_onnx2ncnn bin.')
     parser.add_argument(
-        '--repo-dir', type=str, default='~/', help='mmcls directory.')
+        '--repo-dir', type=str, default='~/', help='mmpretrain directory.')
     parser.add_argument(
         '--out',
         type=str,
@@ -56,7 +56,7 @@ def generate_onnx(args):
         model = conf[1]
         convert_cmd = [
             'python', 'tools/deploy.py',
-            'configs/mmcls/classification_ncnn_static.py', config, model,
+            'configs/mmpretrain/classification_ncnn_static.py', config, model,
             'cat-dog.png', '--work-dir', 'work_dir', '--device', 'cpu'
         ]
         print(subprocess.call(convert_cmd))

.github/scripts/windows/test_full_pipeline.ps1

@@ -23,16 +23,16 @@ $work_dir="work_dir"
 New-Item -Path $work_dir, .\data -ItemType Directory -Force
 $model_cfg="$work_dir\resnet18_8xb32_in1k.py"
 $checkpoint="$work_dir\resnet18_8xb32_in1k_20210831-fbbb1da6.pth"
-$sdk_cfg="configs\mmcls\classification_sdk_dynamic.py"
+$sdk_cfg="configs\mmpretrain\classification_sdk_dynamic.py"
 $input_img="tests\data\tiger.jpeg"
 
-python -m mim download mmcls --config resnet18_8xb32_in1k --dest $work_dir
+python -m mim download mmpretrain --config resnet18_8xb32_in1k --dest $work_dir
 
 if ($Backend -eq "ort") {
-    $deploy_cfg="configs\mmcls\classification_onnxruntime_dynamic.py"
+    $deploy_cfg="configs\mmpretrain\classification_onnxruntime_dynamic.py"
     $model="$work_dir\end2end.onnx"
 } elseif ($Backend -eq "trt") {
-    $deploy_cfg="configs\mmcls\classification_tensorrt-fp16_dynamic-224x224-224x224.py"
+    $deploy_cfg="configs\mmpretrain\classification_tensorrt-fp16_dynamic-224x224-224x224.py"
     $model="$work_dir\end2end.engine"
 } else {
     Write-Host "Unsupported Backend=$Backend"

.github/workflows/backend-ort.yml

@@ -37,9 +37,9 @@ jobs:
           python -m pip install mmcv-lite openmim
           python tools/scripts/build_ubuntu_x64_ort.py 8
           python -c 'import mmdeploy.apis.onnxruntime as ort_api; assert ort_api.is_available(with_custom_ops=True)'
-      - name: test mmcls full pipeline
+      - name: test mmpretrain full pipeline
         run: |
-          python -m mim install $(cat requirements/codebases.txt | grep mmcls)
+          python -m mim install $(cat requirements/codebases.txt | grep mmpretrain)
           export MMDEPLOY_DIR=$(pwd)
           export ONNXRUNTIME_DIR=$MMDEPLOY_DIR/../mmdeploy-dep/onnxruntime-linux-x64-1.8.1
           export LD_LIBRARY_PATH=$ONNXRUNTIME_DIR/lib:$MMDEPLOY_DIR/build/install/lib:$LD_LIBRARY_PATH

.github/workflows/backend-pplnn.yml

@@ -33,7 +33,7 @@ jobs:
           python3 -V
           python3 -m pip install openmim numpy
           python3 -m pip install -r requirements.txt
-          python3 -m mim install $(cat requirements/codebases.txt | grep mmcls)
+          python3 -m mim install $(cat requirements/codebases.txt | grep mmpretrain)
           python3 -m pip list
       - name: Build SDK
         run: |

.github/workflows/build.yml

@@ -233,7 +233,7 @@ jobs:
           python3 -m pip install opencv-python==4.5.4.60 opencv-python-headless==4.5.4.60 opencv-contrib-python==4.5.4.60
           python3 -m pip install openmim numpy pycuda clip transformers
           python3 -m pip install -r requirements.txt
-          python3 -m mim install $(cat requirements/codebases.txt | grep mmcls)
+          python3 -m mim install $(cat requirements/codebases.txt | grep mmpretrain)
           python3 -m pip list
       - name: Build SDK
         run: |
@@ -281,7 +281,7 @@ jobs:
           python -m pip install openmim
           python -m pip install -r requirements.txt
           python -m pip install -r requirements/backends.txt
-          python -m mim install "mmcls>=1.0.0rc2"
+          python -m mim install "mmpretrain>=1.0.0rc7"
           python -m pip list
       - name: Build mmdeploy
         run: |

.github/workflows/quantize.yml

@@ -45,12 +45,12 @@ jobs:
           python -m pip install -r requirements.txt
           python -m pip install -U numpy
 
-      - name: Install mmcls
+      - name: Install mmpretrain
         run: |
           python -m pip install openmim
-          python -m mim install $(cat requirements/codebases.txt | grep mmcls)
-          git clone --depth 1 --branch 1.x https://github.com/open-mmlab/mmclassification.git ~/mmclassification
-          cd ~/mmclassification && python -m pip install . && cd -
+          python -m mim install $(cat requirements/codebases.txt | grep mmpretrain)
+          git clone --depth 1 --branch 1.x https://github.com/open-mmlab/mmpretrain.git ~/mmpretrain
+          cd ~/mmpretrain && python -m pip install . && cd -
       - name: Install ppq
         run: |
           python -m pip install protobuf==3.20.0

README.md

@@ -1,6 +1,6 @@
 _base_ = ['../_base_/torchscript_config.py', '../_base_/backends/coreml.py']
 
-codebase_config = dict(type='mmcls', task='Classification')
+codebase_config = dict(type='mmpretrain', task='Classification')
 
 backend_config = dict(model_inputs=[
     dict(

README_zh-CN.md

@@ -4,4 +4,4 @@
 
 backend_config = dict(
     pipeline=[dict(type='LoadImageFromFile'),
-              dict(type='PackClsInputs')])
+              dict(type='PackInputs')])

configs/mmcls/classification_static.py

@@ -0,0 +1,3 @@
+_base_ = ['../_base_/onnx_config.py']
+
+codebase_config = dict(type='mmpretrain', task='Classification')

configs/mmcls/classification_ascend_dynamic-224x224-224x224.py configs/mmpretrain/classification_ascend_dynamic-224x224-224x224.py

@@ -3,4 +3,4 @@
 ]
 
 ir_config = dict(input_shape=None)
-codebase_config = dict(type='mmcls', task='Classification')
+codebase_config = dict(type='mmpretrain', task='Classification')

configs/mmcls/classification_ascend_static-224x224.py configs/mmpretrain/classification_ascend_static-224x224.py

@@ -18,5 +18,10 @@ endif ()
 
 foreach (codebase IN LISTS CODEBASES)
     message(STATUS "build codebase: ${codebase}")
-    add_subdirectory(${codebase})
+    if (codebase STREQUAL "mmpretrain")
+        set(subdir_name "mmcls")
+    else()
+	set(subdir_name ${codebase})
+    endif()
+    add_subdirectory(${subdir_name})
 endforeach ()