SkyTNT / anime-segmentation

high-accuracy segmentation for anime character
Apache License 2.0
649 stars 62 forks source link

How should the data for training look like? #9

Closed Tentenn closed 1 year ago

Tentenn commented 1 year ago

I have hand annotated segmentation data. (image - mask pairs in png format) and want to train the model to make it better

I see that the training scripts makes use of 4 folders: fg/, bg/, masks/, imgs/

What images go into each folder?

Are there any constraints to the data? (pixel size, image size)

Thank you for your work!

SkyTNT commented 1 year ago

You're almost right.

Did you check out my dataset https://huggingface.co/datasets/skytnt/anime-segmentation ?

Tentenn commented 1 year ago

Thanks! I checked it out now. It makes more sense now.