DevashishPrasad / CascadeTabNet

This repository contains the code and implementation details of the CascadeTabNet paper "CascadeTabNet: An approach for end to end table detection and structure recognition from image-based documents"
MIT License
1.46k stars 422 forks source link

Not detecting every cells #158

Open sanjaychamlagain123 opened 2 years ago

sanjaychamlagain123 commented 2 years ago

I finetuned the model from epoch_36 to epoch_78.. I have 3 class border, borderless and cells..Using transfer learning from provided checkpoint 36 and fine tuning upto 78, i can accurately classify bordered and borderless table but model misses cells detection by 50%. I have trained on custom financial dataset 18k data samples... Can you provide some techniques inorder to detect cells accurately.. should i contime training few more epochs?? Should i tune some specific parameters so i can detect all cells???

Screen Shot 2022-02-17 at 22 02 42
fidhaph commented 1 year ago

@sanjaychamlagain123 could you please provide the code used for finetuning.