Procházet zdrojové kódy

Fix restoration bugs

Bobholamovic před 2 roky
rodič
revize
f0a43b9e0e
2 změnil soubory, kde provedl 2 přidání a 2 odebrání
  1. 1 1
      docs/intro/model_zoo.md
  2. 1 1
      paddlers/tasks/restorer.py

+ 1 - 1
docs/intro/model_zoo.md

@@ -25,7 +25,7 @@ PaddleRS目前已支持的全部模型如下(标注\*的为遥感专用模型
 | 场景分类 | MobileNetV3 | 否 |
 | 场景分类 | MobileNetV3 | 否 |
 | 场景分类 | ResNet50-vd | 否 |
 | 场景分类 | ResNet50-vd | 否 |
 | 图像复原 | DRN | 否 |
 | 图像复原 | DRN | 否 |
-| 图像复原 | ESRGAN |  |
+| 图像复原 | ESRGAN |  |
 | 图像复原 | LESRCNN | 否 |
 | 图像复原 | LESRCNN | 否 |
 | 目标检测 | Faster R-CNN | 否 |
 | 目标检测 | Faster R-CNN | 否 |
 | 目标检测 | PP-YOLO | 否 |
 | 目标检测 | PP-YOLO | 否 |

+ 1 - 1
paddlers/tasks/restorer.py

@@ -773,7 +773,7 @@ class LESRCNN(BaseRestorer):
                  group=1,
                  group=1,
                  **params):
                  **params):
         params.update({
         params.update({
-            'scale': sr_factor,
+            'scale': sr_factor if sr_factor is not None else 1,
             'multi_scale': multi_scale,
             'multi_scale': multi_scale,
             'group': group
             'group': group
         })
         })