def generateBBox():
rootdir = "/home/hanson/dataset/CelebA/Img/img_celeba.7z/img_celeba"
label_landmark_file = "/home/hanson/dataset/CelebA/Anno/list_landmarks_celeba.txt"
label_rect_file = "/home/hanson/dataset/CelebA/Anno/list_bbox_celeba.txt"
imgpath_list = []
landmark_list = []
rect_list = []
label_landmark_frame = pd.read_csv(label_landmark_file,
delim_whitespace=True,
header=None)
for infor in label_landmark_frame.iterrows():
imgpath = os.path.join(rootdir, infor[1][0])
imgpath_list.append(imgpath)
landmark_5p = infor[1][1:11].values.astype(np.int).reshape(
(-1, 2)).tolist()
landmark_list.append(landmark_5p)
label_rect_frame = pd.read_csv(label_rect_file,
delim_whitespace=True,
header=None)
for infor in label_rect_frame.iterrows():
imgpath = os.path.join(rootdir, infor[1][0])
rect = infor[1][1:5].values.astype(np.int).tolist()
rect_list.append(rect)
for imgpath, rect, landmark in zip(imgpath_list, rect_list, landmark_list):
yield imgpath, fu.rect(rect), landmark, None
def generateBBox():
rootdir = "/home/hanson/dataset/CelebA/Img/img_celeba.7z/img_celeba"
labelfile = "label/img_celeba_facepp_label.csv"
label_frame = pd.read_csv(labelfile, sep=" ")
for infor in label_frame.iterrows():
imgpath = os.path.join(rootdir, infor[1][0])
rect = fu.rect(infor[1][1:5].values.astype(np.int).tolist())
landmark_106p = infor[1][8:220].values.astype(np.int).reshape(
(106, 2)).tolist()
landmark_5p = [landmark_106p[i] for i in [75, 85, 54, 86, 91]]
yield imgpath, rect, landmark_5p, landmark_106p
def generateBBox():
rootdir = "/home/hanson/dataset/landmark/menpo/menpo_train_release"
labelfile = "/home/hanson/work/FaceLandmark_Pytorch/dataset/label/menpo_train_release_facepp_label.csv"
label_frame = pd.read_csv(labelfile, sep=" ")
for infor in label_frame.iterrows():
imgpath = os.path.join(rootdir, infor[1][0])
rect = fu.rect(infor[1][1:5].values.astype(np.int).tolist())
landmark_106p = infor[1][8:220].values.astype(np.int).reshape(
(106, 2)).tolist()
landmark_5p = [landmark_106p[i] for i in [75, 85, 54, 86, 91]]
landmark_6p = [landmark_106p[i] for i in [75, 85, 54, 86, 91, 16]]
yield imgpath, rect, landmark_5p, landmark_6p, landmark_106p
def generateBBox():
rootdir = "/home/hanson/work/FaceLandmark_Pytorch/dataset/data/WFLW/WFLW_images"
labelfile = "/home/hanson/work/FaceLandmark_Pytorch/dataset/data/WFLW/WFLW_images_facepp_label.csv"
label_frame = pd.read_csv(labelfile, sep=" ")
for infor in label_frame.iterrows():
imgpath = os.path.join(rootdir, infor[1][0])
rect = fu.rect(infor[1][1:5].values.astype(np.int).tolist())
landmark = infor[1][8:220].values.astype(np.int).reshape(
(106, 2)).tolist()
new_landmark = []
#print (landmark[75])
new_landmark.append(landmark[75])
new_landmark.append(landmark[85])
new_landmark.append(landmark[54])
new_landmark.append(landmark[86])
new_landmark.append(landmark[91])
yield imgpath, rect, new_landmark