[Enhancement] support all matting models inferencer (open-mmlab#1836)

Author: Z-Fran

demo/README.md

@@ -124,9 +124,9 @@ python mmagic_inference_demo.py \
 
 ```shell
 python mmagic_inference_demo.py \
-        --model-name global_local  \
+        --model-name gca  \
         --img ../resources/input/matting/GT05.jpg \
-        --mask ../resources/input/matting/GT05_trimap.jpg \
+        --trimap ../resources/input/matting/GT05_trimap.jpg \
         --result-out-dir ../resources/output/matting/demo_matting_gca_res.png
 ```
 

mmagic/apis/inferencers/matting_inferencer.py

@@ -7,8 +7,6 @@
 import torch
 from mmengine import mkdir_or_exist
 from mmengine.dataset import Compose
-from mmengine.dataset.utils import default_collate as collate
-from torch.nn.parallel import scatter
 
 from mmagic.structures import DataSample
 from .base_mmagic_inferencer import BaseMMagicInferencer, InputsType, PredType
@@ -53,21 +51,15 @@ def preprocess(self, img: InputsType, trimap: InputsType) -> Dict:
         _data = test_pipeline(data)
         trimap = _data['data_samples'].trimap.data
         preprocess_res = dict()
-        preprocess_res['inputs'] = torch.cat([_data['inputs'], trimap],
-                                             dim=0).float()
-        preprocess_res = collate([preprocess_res])
-        preprocess_res['data_samples'] = DataSample.stack(
-            [_data['data_samples']])
-        preprocess_res['mode'] = 'predict'
-        if 'cuda' in str(self.device):
-            preprocess_res = scatter(preprocess_res, [self.device])[0]
-
+        preprocess_res['inputs'] = [_data['inputs']]
+        preprocess_res['data_samples'] = [_data['data_samples']]
         return preprocess_res
 
     def forward(self, inputs: InputsType) -> PredType:
         """Forward the inputs to the model."""
+        inputs = self.model.data_preprocessor(inputs)
         with torch.no_grad():
-            return self.model(**inputs)
+            return self.model(mode='predict', **inputs)
 
     def visualize(self,
                   preds: PredType,

mmagic/apis/mmagic_inferencer.py

@@ -63,6 +63,8 @@ class MMagicInferencer:
         'styleganv3',
 
         # matting models
+        'dim',
+        'indexnet',
         'gca',
 
         # inpainting models