table-transformer icon indicating copy to clipboard operation
table-transformer copied to clipboard

How to use table detection pre-trained weights to train a custom dataset

Open Aditi81 opened this issue 2 years ago • 2 comments

I have created a custom table detection dataset that has different class labels. Instead of training a model from scratch, I want to use the pre-trained weights for the table detection model trained on PubTables-1M. Please help me to proceed with this task.

Aditi81 avatar Jun 16 '22 07:06 Aditi81

This should be possible with probably just a few changes to the code depending on how closely what you're trying to do matches what we did. You said you have a custom dataset for table detection with different class labels. So you have class labels besides "table" and "table rotated"?

bsmock avatar Jun 16 '22 16:06 bsmock

I have different types of tables as a class.

In addition, how should I use the pre-trained weights for the table detection model trained on PubTables-1M?

Aditi81 avatar Jun 16 '22 17:06 Aditi81