diff --git a/hand_data_iter/datasets.py b/hand_data_iter/datasets.py index e6e2cd4f131b7a1e3b2d78fe778ae23a45643478..88898fa147bcbcf03b62ceec5ee6406fbc923b8f 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)