traing dataset

[4ff4n]

hello there! thank you so much for sharing your amazing work. would you please assist me how can i prepare my own custom dataset and what should the the directory structure in order to call the main.py to start the training process?

thanks Affan

[fdbtrs]

You should create one folder (e.g. id1, id2.....idn) for each identity in your dataset. Inside each folder, you should create two folders original (embedding of unmasked images)and fakemask (embedding of masked images)