How to train edge to face on my own data

Thanks for the great work! I wonder if i have a face dataset, what do i need to do to prepare it for edge to face training. Do i have to extract segments for all face parts like CelebAMask?