GFPGAN icon indicating copy to clipboard operation
GFPGAN copied to clipboard

如何生成自己数据的landmark文件?

Open alexliyang opened this issue 3 years ago • 5 comments

FFHQ_eye_mouth_landmarks_512.pth 这个是FFHQ的数据集的landmark,如果要训练自己的数据集,这个如何生成?比如用CeleBA 数据集

alexliyang avatar Nov 02 '21 09:11 alexliyang

你可以在inference的代码里找找 里面肯定有提取提取输入图片的landmark的部分 我之前看到过 具体不愿找了。反正是用到了facexlib这个库里的函数

jorjiang avatar Nov 04 '21 05:11 jorjiang

I found it on scripts/parse_landmark.py

but the face_landmarks from ffhq-dataset-v2.json, so you need to get face_landmarks by dlib for your datasets

onefish51 avatar Jul 25 '22 09:07 onefish51

您找到方法了吗

BrotherQi avatar Aug 27 '22 08:08 BrotherQi

you can make it by dlib or face_alignment

onefish51 avatar Aug 31 '22 07:08 onefish51

firstly thanks for your reply, my error is AttributeError: 'GFPGANModel' object has no attribute 'loc_left_eyes' but i dont see the same key 'loc_left_eyes' in scripts/parse_landmark.py can you explain it in detail

BrotherQi avatar Aug 31 '22 09:08 BrotherQi