clean code

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,

ppgan/apps/__init__.py

+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

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,