Fix restoration bugs

2 years ago · f0a43b9e0e
parent 92a5086c79
commit f0a43b9e0e
2 changed files with 2 additions and 2 deletions
--- a/docs/intro/model_zoo.md
+++ b/docs/intro/model_zoo.md
@ -25,7 +25,7 @@ PaddleRS目前已支持的全部模型如下（标注\*的为遥感专用模型
 | 场景分类 | MobileNetV3 | 否 |
 | 场景分类 | ResNet50-vd | 否 |
 | 图像复原 | DRN | 否 |
-| 图像复原 | ESRGAN | 否 |
+| 图像复原 | ESRGAN | 是 |
 | 图像复原 | LESRCNN | 否 |
 | 目标检测 | Faster R-CNN | 否 |
 | 目标检测 | PP-YOLO | 否 |
--- a/paddlers/tasks/restorer.py
+++ b/paddlers/tasks/restorer.py
@ -773,7 +773,7 @@ class LESRCNN(BaseRestorer):
                 group=1,
                 **params):
        params.update({
-            'scale': sr_factor,
+            'scale': sr_factor if sr_factor is not None else 1,
            'multi_scale': multi_scale,
            'group': group
        })