search

faustomorales / keras-ocr

A packaged and flexible version of the CRAFT text detector and Keras CRNN recognition model.
https://keras-ocr.readthedocs.io/
MIT License
1.37k stars 349 forks source link

where does `craft_mlt_25k.h5` come from? #188

Open ghost opened 2 years ago

ghost commented 2 years ago

Hi, thanks for your great work which I'm referencing in my version of craft. I noticed that in the training example your code looks for / downloads a craft_mlt_25k.h5. Assuming I want to train a model from scratch without pretrained weights, after n training epochs, should I expect to get the same weights or close? If not, then how these weights were originally obtained? I also noticed that the training always starts with a loss value of 0.0something whether pretrained weights are loaded or not. This is making me question whether the training is actually working.

Asad-Raza commented 2 years ago

Hi , Did you get answer to the question? Im also interested in knowing the answer for this question.

Does Keras OCR use pretrained model of pyTorch CRAFT implementation?

ghost commented 2 years ago

Hi , Did you get answer to the question? Im also interested in knowing the answer for this question.

Does Keras OCR use pretrained model of pyTorch CRAFT implementation?

Not yet but I'm assuming these weights were converted from pytorch, and clearly there's something wrong with the loss starting at a very low value.