train.py file doesn’t exist

[ssatz]

Hi, thanks for the repo. I am trying to train the pub1m table structure on yolo8. I couldn’t find the train.py file.
Label table row accuracy is no at all good. Any suggestions

[whn09]

I used Yolov5 to train the model, but I think Yolov8 is similar

[ssatz]

hi thanks this has been solved. I looked into wrong Yolo version.
I think you didn't train the Fintab Dataset?

[whn09]

Yes, I didn't train the Fintab Dataset. But maybe you can merge the dataset with PubTables-1M. I also recommend you to set Yolo size to at least 800 to get best performance.

[ssatz]

Yes I just trained for fintab with 640 size and table detection is not good and getting short rows. I will try it with 800.

[ssatz]

I trained both Fintab and PubTables- 1m and normal structure the results are good and i think need to train PaddleOCR for text detection.

Here is the Yolo format data https://huggingface.co/datasets/Codeplug/pub-fintab-yolov
Metrics:

For training the dataset we have used Runpod