From cc55048172aee51660258b49673a7d29f9e94c75 Mon Sep 17 00:00:00 2001
From: "Eric.Lee2021" <305141918@qq.com>
Date: Sun, 21 Feb 2021 22:59:50 +0800
Subject: [PATCH] update

---
 hand_data_iter/datasets.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/hand_data_iter/datasets.py b/hand_data_iter/datasets.py
index e6e2cd4..88898fa 100644
--- a/hand_data_iter/datasets.py
+++ b/hand_data_iter/datasets.py
@@ -163,7 +163,7 @@ class LoadImagesAndLabels(Dataset):
                             pt_right = (x_max+random.randint(-offset_x,offset_x),(y_min+y_max)/2+random.randint(-offset_y,offset_y))
                             angle_random = random.randint(-180,180)
                             scale_x = float(random.randint(20,32))/100.
-                            hand_rot,pts_tor_landmarks = hand_alignment_aug_fun(img_ago,pt_left,pt_right,
+                            hand_rot,pts_tor_landmarks,_ = hand_alignment_aug_fun(img_ago,pt_left,pt_right,
                                 facial_landmarks_n = pts_,\
                                 angle = angle_random,desiredLeftEye=(scale_x, 0.5),
                                 desiredFaceWidth=img_size[0], desiredFaceHeight=None,draw_flag = True)
-- 
GitLab