search

pythonlessons / mltu

Machine Learning Training Utilities (for TensorFlow and PyTorch)
MIT License
160 stars 100 forks source link

Regarding CSV file based datasets for 11_yolov_train for text detection #55

Open Rajaro72 opened 1 month ago

Rajaro72 commented 1 month ago

here is the dataset link: https://www.kaggle.com/datasets/robikscube/textocr-text-extraction-from-images-dataset I was struggling trying to adapt the code for the dataset above, the model would freeze after 1 full epoch

pythonlessons commented 3 weeks ago

I almost integrated feature for TextOCR dataset, I'll reopen this and close after I merge these changes

Rajaro72 commented 3 weeks ago

Thank you! I managed to stop it from freezing at 1 epoch and it ran for 4 epochs, then early stopped, but the model didn't work, I think it was due to the images not being loaded properly, but tried a different approach. I attached my attempts on #56