From 52cc814fec90eda6e0b87ac9dd2e05bb1cc605e2 Mon Sep 17 00:00:00 2001
From: LielinJiang <jianglielin@baidu.com>
Date: Tue, 13 Oct 2020 05:05:16 +0000
Subject: [PATCH] clean code

---
 applications/tools/video-enhance.py  | 8 ++------
 ppgan/apps/__init__.py               | 5 +++++
 ppgan/apps/deepremaster_predictor.py | 2 +-
 3 files changed, 8 insertions(+), 7 deletions(-)
 create mode 100644 ppgan/apps/__init__.py

diff --git a/applications/tools/video-enhance.py b/applications/tools/video-enhance.py
index 6aea857..b5a9b02 100644
--- a/applications/tools/video-enhance.py
+++ b/applications/tools/video-enhance.py
@@ -15,11 +15,7 @@
 import argparse
 import paddle
 
-from ppgan.apps.dain_predictor import DAINPredictor
-from ppgan.apps.deepremaster_predictor import DeepReasterPredictor
-from ppgan.apps.deoldify_predictor import DeOldifyPredictor
-from ppgan.apps.realsr_predictor import RealSRPredictor
-from ppgan.apps.edvr_predictor import EDVRPredictor
+from ppgan.apps import DAINPredictor, DeepRemasterPredictor, DeOldifyPredictor, RealSRPredictor, EDVRPredictor
 
 parser = argparse.ArgumentParser(description='Fix video')
 parser.add_argument('--input', type=str, default=None, help='Input video')
@@ -97,7 +93,7 @@ if __name__ == "__main__":
             frames_path, temp_video_path = predictor.run(temp_video_path)
             paddle.disable_static()
         elif order == 'DeepRemaster':
-            predictor = DeepReasterPredictor(
+            predictor = DeepRemasterPredictor(
                 args.output,
                 weight_path=args.DeepRemaster_weight,
                 colorization=args.colorization,
diff --git a/ppgan/apps/__init__.py b/ppgan/apps/__init__.py
new file mode 100644
index 0000000..3dca0cc
--- /dev/null
+++ b/ppgan/apps/__init__.py
@@ -0,0 +1,5 @@
+from .dain_predictor import DAINPredictor
+from .deepremaster_predictor import DeepRemasterPredictor
+from .deoldify_predictor import DeOldifyPredictor
+from .realsr_predictor import RealSRPredictor
+from .edvr_predictor import EDVRPredictor
diff --git a/ppgan/apps/deepremaster_predictor.py b/ppgan/apps/deepremaster_predictor.py
index a63fa16..596abe6 100644
--- a/ppgan/apps/deepremaster_predictor.py
+++ b/ppgan/apps/deepremaster_predictor.py
@@ -64,7 +64,7 @@ def addMergin(img, target_w, target_h, background_color=(0, 0, 0)):
     return result
 
 
-class DeepReasterPredictor(BasePredictor):
+class DeepRemasterPredictor(BasePredictor):
     def __init__(self,
                  output='output',
                  weight_path=None,
-- 
GitLab