u-2-net-portrait icon indicating copy to clipboard operation
u-2-net-portrait copied to clipboard

What dataset did you use?

Open dfrumkin opened this issue 4 years ago • 3 comments

Hello Dennis! What dataset did you use for training?

dfrumkin avatar Apr 25 '21 13:04 dfrumkin

Mainly StyleGAN synthesized portraits, the supervisely person dataset and the aisegment dataset, however it is important to highlight that the model is not fully trained just for a couple of epochs to validate the approach. Synthesized data contains a lot of noisy labels e.g. mislabeled fine elements however overall it yielded interesting results.

dbpprt avatar Apr 26 '21 18:04 dbpprt

@dennisbappert how are you generating the high-quality masks for training for the StyleGAN portraits?

mln00b avatar Apr 27 '21 15:04 mln00b

@dennisbappert If you can share the dataset with high quality masks that would really be helpfull. mail id: [email protected]

hackkhai avatar Sep 13 '21 13:09 hackkhai