search

rinkstiekema / table-structures

87 stars 30 forks source link

Is it possible to retrain on custom Table data #10

Closed Sharathmk99 closed 4 years ago

Sharathmk99 commented 4 years ago

Hi. Thank you for the great work and open sourcing the same. I was wondering is it possible to train from scratch and also retrain on custom Table data? Thank you.

rinkstiekema commented 4 years ago

Absolutely, you can take any of the models and retrain them using train.py in: ./pipeline/pix2pixHD/train.py and ./pipeline/segmentation/train.py

You could also train any other type of model if you want.

Sharathmk99 commented 4 years ago

@rinkstiekema Can you share sample training data required to train the model? How should we annotate the table?

rinkstiekema commented 4 years ago

You can find an example dataset in this google drive: https://drive.google.com/drive/u/0/folders/1dgKISbhBNfR8XXnIxUD_sIhwYNurKHbb