More diversity in the dataset

[Conobi]

As the data comes from Baidu, there is a large amount of Asian people (particularly Chinese couples). It would be a good idea to integrate and train the model with more ethnically diverse couples in order to have better results for others.

[irfanICMLL]

Thank you for your advice~

If anyone is interested, he or she can use the training code I provided.

[Conobi]

Has anyone any good dataset with multi-ethnicals couples?