fine-tuning on a custom dataset with the official craft-pytorch weights

[alikaz3mi]

Hello, is it possible to fine-tune your model on a custom dataset with Craft-PyTorch released weights too?

[gl94]

Yes, I just made it using the IC15 traing code. I labeled my dataset the same format as IC15.

[tahakoudri]

Hello, gl94. Can you explain how did you manage do to it and if it works well ?

[gl94]

Hello, gl94. Can you explain how did you manage do to it and if it works well ?

It's kinda a little bit tricky, you need to use the old branch of this repo, and change some code in the trainic15data.py and maybe some function in the dataloader to get a better model. After all this been done you'll find that craft is really hard to train. I rather recommend you using dbnet in the PPocr. It's easy to train and performs very well.

[SkygirlLuna]

@gl94 What is PPocr?