search

Bartzi / stn-ocr

Code for the paper STN-OCR: A single Neural Network for Text Detection and Text Recognition
https://arxiv.org/abs/1707.08831
GNU General Public License v3.0
499 stars 137 forks source link

load pretrained model error #34

Open fycfycfyc opened 3 years ago

fycfycfyc commented 3 years ago

Hello, I encountered another problem when loading the pre-trained model, as shown in the following figure: 微信图片_20210809205306 微信图片_20210809205256

When calling the python svhn_train.py --model_prefix provided by you, the responding model file cannot always be found, but I switched the directory to this directory and found that there is a responding model file in this directory, so it is strange. The only difference between me and you is that my mxnet version is 1.0.0 instead of 0.9.3, but I think the functions of the two versions of the model load should be the same, and it will not cause the error. In addition, I would like to ask you, is there a difference between see-ocr and stn-ocr? Are the two models exactly the same? What is the difference between the two?

Looking forward to your reply！

Bartzi commented 3 years ago

think you have to provide the full name of the params file. Did you try that already?

The main difference between SEE and STN-OCR is that SEE is written in Chainer (which is simpler to use) and has some different code. However, all in all they are quite similar, I think. I wrote that code about 4 years ago :sweat_smile:.