zhang0jhon/AttentionOCR

Train my own data

Opened this issue · 2 comments

Hello, can I train the data set I collected? If yes, how can I label the data and train

And I want to know the label tool

The dataset.py file is where you want to look. I modified the LSVT class to load in my dataset (downloading that dataset helped), and then ran the last few lines of the dataset.py script to output my new class to the dataset file instead of joining all of them together.

However, I am having issues with training now due to the age the libraries being used. I'm still trying to figure that out.