diff --git a/docs/intro/model_zoo.md b/docs/intro/model_zoo.md index f8b520d..faf7cd2 100644 --- 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 | 否 | diff --git a/paddlers/tasks/restorer.py b/paddlers/tasks/restorer.py index ce7e931..ff8708b 100644 --- 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 })