fix namespace bug.

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~~
 

inference_codeformer.py

@@ -57,7 +57,7 @@ if __name__ == '__main__':
     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 @@ if __name__ == '__main__':
     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]