-
Notifications
You must be signed in to change notification settings - Fork 448
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
Train (fine-tune) LayoutParser pre-trained model using custom dataset #177
Comments
I have the exact same question |
Hi @Ammar-Azman and @epassaro, You have this tutorial that explains how to fine-tune the models available on the model zoo: https://towardsdatascience.com/auto-parse-and-understand-any-document-5d72e81b0be9 The model weights link is within the |
Thank you, I already achieved that using the Detectron2 tutorial, but I will take a look anyway. |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Motivation
I want to fine-tune the pre-train model from LayoutParser by using my own dataset images. But I don't find any way in the documentation where I can train the pre-train model using my custom dataset, instead I only can train the Fast_RCNN or MaskRCNN model which means training from scratch. This might required million of dataset to get to get better performance metrics.
Is there any way I can fine-tune the model from LayoutParser? Is this even possible?
The text was updated successfully, but these errors were encountered: