[Enhance] support image restoration inferencer

demo/README.md

@@ -30,6 +30,10 @@ Table of contents:
 
        [2.2.9. Text-to-Image example](#229-text-to-image)
 
+       [2.2.10. 3D-aware Generation example](#2210-3d-aware-generation)
+
+       [2.2.11. Image Restoration example](#2211-image-restoration)
+
 [3. Other demos](#3-other-demos)
 
 ## 1. Download sample images or videos
@@ -214,10 +218,19 @@ python mmagic_inference_demo.py \
         --result-out-dir ../resources/output/text2image/demo_text2image_stable_diffusion_res.png
 ```
 
-#### 2.2.9 3D-aware Generation (EG3D)
+#### 2.2.10 3D-aware Generation
 
 ```shell
 python demo/mmagic_inference_demo.py \
     --model-name eg3d \
     --result-out-dir ../resources/output/eg3d-output
 ```
+
+#### 2.2.11 Image Restoration
+
+```shell
+python mmagic_inference_demo.py \
+        --model-name nafnet \
+        --img ../resources/input/restoration/0901x2.png \
+        --result-out-dir ../resources/output/restoration/demo_restoration_nafnet_res.png
+```

mmagic/apis/inferencers/__init__.py

@@ -84,6 +84,13 @@ def __init__(self,
                 config, ckpt, device, extra_parameters, seed=seed)
         elif self.task in ['controlnet_animation']:
             self.inferencer = ControlnetAnimationInferencer(config)
+        elif self.task in [
+                'Image Restoration', 'Denoising, Deblurring, Deraining',
+                'Image Super-Resolution, Image denoising, JPEG compression '
+                'artifact reduction'
+        ]:
+            self.inferencer = ImageSuperResolutionInferencer(
+                config, ckpt, device, extra_parameters, seed=seed)
         else:
             raise ValueError(f'Unknown inferencer task: {self.task}')
 

mmagic/apis/mmagic_inferencer.py

@@ -101,6 +101,11 @@ class MMagicInferencer:
         'basicvsr_pp',
         'real_basicvsr',
 
+        # image_restoration models
+        'nafnet',
+        'swinir',
+        'restormer',
+
         # text2image models
         'disco_diffusion',
         'stable_diffusion',

requirements/runtime.txt

@@ -7,6 +7,7 @@ face-alignment
 facexlib
 lmdb
 lpips
+mediapipe
 mmcv>=2.0.0rc1
 mmengine
 numpy