fix namespace bug.

xxmiprai · Oct 9, 2022 · fa547b7 · fa547b7
1 parent 3014b64
commit fa547b7
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -36,6 +36,7 @@ S-Lab, Nanyang Technological University
 
 ### TODO
 - [ ] Add checkpoint for face inpainting
+- [ ] Add checkpoint for face colorization
 - [ ] Add training code and config files
 - [x] ~~Add background image enhancement~~
 

diff --git a/inference_codeformer.py b/inference_codeformer.py
@@ -57,7 +57,7 @@ def set_realesrgan():
     parser.add_argument('-o', '--output_path', type=str, default=None, 
             help='Output folder. Default: results/<input_name>_<w>')
     parser.add_argument('-w', '--fidelity_weight', type=float, default=0.5, 
-            help='Balance the quality and fidelity')
+            help='Balance the quality and fidelity. Default: 0.5')
     parser.add_argument('-s', '--upscale', type=int, default=2, 
             help='The final upsampling scale of the image. Default: 2')
     parser.add_argument('--has_aligned', action='store_true', help='Input are cropped and aligned faces. Default: False')
@@ -77,7 +77,7 @@ def set_realesrgan():
     args = parser.parse_args()
 
     # ------------------------ input & output ------------------------
-    w = args.w
+    w = args.fidelity_weight
     input_video = False
     if args.input_path.endswith(('jpg', 'png')): # input single img path
         input_img_list = [args.input_path]