search

Bartzi / see

Code for the AAAI 2018 publication "SEE: Towards Semi-Supervised End-to-End Scene Text Recognition"
GNU General Public License v3.0
573 stars 147 forks source link

svhn demo #75

Closed keekeecaca closed 5 years ago

keekeecaca commented 5 years ago

I have a trained SVHN with my custom dataset, any svhn_demo.py or guidance for svhn inference? thanks.

Bartzi commented 5 years ago

Hmm, in the end it should not be that different from the other two demo files. You just need to load the model and initialize the weights of the chain with the trained model. Then you'll have to load an image and convert it the same way, you converted your training images. After you've done that you just call the chain and do a forward pass with your input image. You will get the bounding box information and the predeicted characters, and then you can do whatever you like with those.

So just have a look at the other two demo scripts. You should not need many adjustments.

ahmed-nady commented 3 years ago

@Bartzi Thanks for sharing your code. After I created a svhn_demo.py as mentioned above. I encountered an issue related to pre-trained models. So, Where Can i find the SVHN model for your 3 experiments or should I train in order to obtain them?

Bartzi commented 3 years ago

Currently, we did not publish the SVHN models. I'm actually not sure whether I can find them again :sweat_smile: it has been quite some time since we did the last experiments. So, your best option is to run the training for each of the models again.