fix typos

datathehuman · Feb 26, 2024 · 3437464 · 3437464
1 parent 8413732
commit 3437464
Show file tree

Hide file tree

Showing 6 changed files with 6 additions and 11 deletions.
diff --git a/.DS_Store b/.DS_Store
diff --git a/README.md b/README.md
@@ -144,7 +144,7 @@ face_image = load_image("./examples/yann-lecun_resize.jpg")
 
 # prepare face emb
 face_info = app.get(cv2.cvtColor(np.array(face_image), cv2.COLOR_RGB2BGR))
-face_info = sorted(face_info, key=lambda x:(x['bbox'][2]-x['bbox'][0])*x['bbox'][3]-x['bbox'][1])[-1]  # only use the maximum face
+face_info = sorted(face_info, key=lambda x:(x['bbox'][2]-x['bbox'][0])*(x['bbox'][3]-x['bbox'][1]))[-1]  # only use the maximum face
 face_emb = face_info['embedding']
 face_kps = draw_kps(face_image, face_info['kps'])
 

diff --git a/gradio_demo/app-multicontrolnet.py b/gradio_demo/app-multicontrolnet.py
@@ -357,12 +357,7 @@ def generate_image(
                 f"Unable to detect a face in the image. Please upload a different photo with a clear face."
             )
 
-        face_info = sorted(
-            face_info,
-            key=lambda x: (x["bbox"][2] - x["bbox"][0]) * x["bbox"][3] - x["bbox"][1],
-        )[
-            -1
-        ]  # only use the maximum face
+        face_info = sorted(face_info, key=lambda x:(x['bbox'][2]-x['bbox'][0])*(x['bbox'][3]-x['bbox'][1]))[-1]  # only use the maximum face
         face_emb = face_info["embedding"]
         face_kps = draw_kps(convert_from_cv2_to_image(face_image_cv2), face_info["kps"])
         img_controlnet = face_image

diff --git a/gradio_demo/app.py b/gradio_demo/app.py
@@ -239,7 +239,7 @@ def generate_image(face_image_path, pose_image_path, prompt, negative_prompt, st
         if len(face_info) == 0:
             raise gr.Error(f"Cannot find any face in the image! Please upload another person image")
 
-        face_info = sorted(face_info, key=lambda x:(x['bbox'][2]-x['bbox'][0])*x['bbox'][3]-x['bbox'][1])[-1]  # only use the maximum face
+        face_info = sorted(face_info, key=lambda x:(x['bbox'][2]-x['bbox'][0])*(x['bbox'][3]-x['bbox'][1]))[-1]  # only use the maximum face
         face_emb = face_info['embedding']
         face_kps = draw_kps(convert_from_cv2_to_image(face_image_cv2), face_info['kps'])
 

diff --git a/infer.py b/infer.py
@@ -64,7 +64,7 @@ def resize_img(input_image, max_side=1280, min_side=1024, size=None,
     face_image = resize_img(face_image)
 
     face_info = app.get(cv2.cvtColor(np.array(face_image), cv2.COLOR_RGB2BGR))
-    face_info = sorted(face_info, key=lambda x:(x['bbox'][2]-x['bbox'][0])*x['bbox'][3]-x['bbox'][1])[-1] # only use the maximum face
+    face_info = sorted(face_info, key=lambda x:(x['bbox'][2]-x['bbox'][0])*(x['bbox'][3]-x['bbox'][1]))[-1] # only use the maximum face
     face_emb = face_info['embedding']
     face_kps = draw_kps(face_image, face_info['kps'])
 

diff --git a/infer_full.py b/infer_full.py
@@ -79,7 +79,7 @@ def resize_img(input_image, max_side=1280, min_side=1024, size=None,
     face_image = resize_img(face_image)
 
     face_info = app.get(cv2.cvtColor(np.array(face_image), cv2.COLOR_RGB2BGR))
-    face_info = sorted(face_info, key=lambda x:(x['bbox'][2]-x['bbox'][0])*x['bbox'][3]-x['bbox'][1])[-1] # only use the maximum face
+    face_info = sorted(face_info, key=lambda x:(x['bbox'][2]-x['bbox'][0])*(x['bbox'][3]-x['bbox'][1]))[-1] # only use the maximum face
     face_emb = face_info['embedding']
 
     # use another reference image
@@ -88,7 +88,7 @@ def resize_img(input_image, max_side=1280, min_side=1024, size=None,
 
     face_info = app.get(cv2.cvtColor(np.array(pose_image), cv2.COLOR_RGB2BGR))
     pose_image_cv2 = convert_from_image_to_cv2(pose_image)
-    face_info = sorted(face_info, key=lambda x:(x['bbox'][2]-x['bbox'][0])*x['bbox'][3]-x['bbox'][1])[-1] # only use the maximum face
+    face_info = sorted(face_info, key=lambda x:(x['bbox'][2]-x['bbox'][0])*(x['bbox'][3]-x['bbox'][1]))[-1] # only use the maximum face
     face_kps = draw_kps(pose_image, face_info['kps'])
 
     width, height = face_kps.size