Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

How to fine tune the detector and the recogniser for a custom dataset? #194

Open
Pusse-01 opened this issue Feb 4, 2022 · 2 comments
Open

Comments

@Pusse-01
Copy link

Pusse-01 commented Feb 4, 2022

Can anyone explain what is the format of the data or how is the dataset should be feeded to the model to fine tune. I have a set of images which have been labeled each and every character by drawing bounding boxes. So how I could fine tune the model using this data set to increase the accuracy of my model.

@JackSadigov
Copy link

Hi.
I am also looking for the same here.
I tried Convert PASCAL dataset to TFRecord for object detection in TensorFlow explained here but got an error with ''annotations has no object".

@seansteel3
Copy link

@Pusse-01 and @JackSadigov here is an excellent guide that will take you through retraining: https://keras-ocr.readthedocs.io/en/latest/examples/end_to_end_training.html

There are two open issues related to:
Detector Batch size: #218
Recognizer retraining: #231

For the recognizer retraining Oadon's solution worked for me, I did not attempt geo-tp's but it probably works too. Obviously make sure you're data is in the same format as the guide (or edit the guide accordingly) to retrain.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

3 participants