MangaLineExtraction_PyTorch icon indicating copy to clipboard operation
MangaLineExtraction_PyTorch copied to clipboard

Training Code

Open giantstonex7 opened this issue 3 years ago • 2 comments

Hello, as you mentioned the training code would be ready soon, what's the progress so far, may i help?

giantstonex7 avatar May 30 '22 03:05 giantstonex7

同上,有训练代码嘛

zccoder avatar Sep 01 '22 08:09 zccoder

The training and tone filling code was written like 6 years ago so there might be some problems when running. We are not maintaining it anymore so please try solving these issues yourself. For line drawings, you can gather your own dataset by searching "line illustrations" or something. We cannot share that to you so sorry. But any common line drawings can produce quite promising results. We only include the filling of screentones in the following link. It's badly written in python2 so I guess you may need some adoption if you need to generate on-the-fly.

DL: Erika.zip

BTW the secret santa dataset is from deviantart but I cannot find the credit.

To train the model you may need to create a PyTorch dataloader yourself with the model code (should be easy).

It includes the essential things for reproducing the model when you have the dataset ready. It's suggested to train with > 5000 tones line drawings. It would be even better if you can include some cherry-picking on samples (by removing the outliers and badly looked samples).

Many thanks.

ljsabc avatar Sep 13 '22 07:09 ljsabc