Norod / U-2-Net-StyleTransfer

U-2-Net: U Square Net - Modified for paired image training of style transfer
https://huggingface.co/spaces/Norod78/ComicsHeroU2Net
Apache License 2.0
49 stars 8 forks source link

Could train the model in other dataset? #2

Open woshildh opened 2 years ago

woshildh commented 2 years ago

I want train the model in other cute dataset. Which dataset format is needed by the model?

Norod commented 2 years ago

@woshildh This repository contains an example of a paired-image training dataset under https://github.com/Norod/U-2-Net-StyleTransfer/tree/main/train_data folder. It's basically two folders with jpg files each pair has the same name