RaidasGrisk / tf2-crnn

TF2 implementation of Text recognition CRNN

TF-2 CRNN

CNN + RNN for Scene Text Recognition implemented using tf2 keras module. Mostly based on this and this repo. The project code is written as simply as I could. This is to make it easy to understand and debug. Also to get to know the tf2 keras module better.

Project structure:

1. Model. The model definition and architecture.
2. Training. Code for training.
3. Test. Code for testing the pre-trained model.
4. Utils. All the utility functions used.

Results

Loss of the pre-trained model.

Some examples of pre-trained model at work.

image	recognition
	Rick
	MORTY
	pontifically
	Sparkling
	Bombastic
	Croons
	Testicles

TODO

• pre-train the model on more data
• visualize training progress