How to train the model on custom dataset?

bloc97 / Anime4K

A High-Quality Real Time Upscaler for Anime Video

https://bloc97.github.io/Anime4K/

MIT License

18.53k stars 1.35k forks source link

How to train the model on custom dataset? #214

Open arianaa30 opened 1 year ago

arianaa30 commented 1 year ago

Is there a guide how to train the model on my own dataset? I noticed the train script under /Tensroflow. But it loads a "/hdd/sdb/SYNLA_Plus_4096.npy" array. How can I generate a similar file with my own dataset? Is there a step-by-step guide how to do that?

Tama47 commented 1 year ago

There isn't a step-by-step guide on how to train the model, the best you might find is from this discussion thread https://github.com/bloc97/Anime4K/discussions/180.