Could train the model in other dataset?

Norod / U-2-Net-StyleTransfer

U-2-Net: U Square Net - Modified for paired image training of style transfer

Apache License 2.0

49 stars 8 forks source link

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