change threshold for is_gray.

facelib/utils/face_restoration_helper.py

@@ -127,7 +127,7 @@ class FaceRestoreHelper(object):
             img = img[:, :, 0:3]
 
         self.input_img = img
-        self.is_gray = is_gray(img, threshold=5)
+        self.is_gray = is_gray(img, threshold=10)
         if self.is_gray:
             print('Grayscale input: True')
 

inference_codeformer.py

@@ -174,7 +174,7 @@ if __name__ == '__main__':
         if args.has_aligned: 
             # the input faces are already cropped and aligned
             img = cv2.resize(img, (512, 512), interpolation=cv2.INTER_LINEAR)
-            face_helper.is_gray = is_gray(img, threshold=5)
+            face_helper.is_gray = is_gray(img, threshold=10)
             if face_helper.is_gray:
                 print('Grayscale input: True')
             face_helper.cropped_faces = [img]