Arthur Flor
1 min readDec 4, 2019

--

Hi!

To run a custom dataset, you must add a function in Dataset class, such as `def _custom_dataset(self)`, where returns a dict variable with:

{‘train’: [‘dt’:[], ‘gt’:[]],

‘valid’: [‘dt’:[], ‘gt’:[]],

‘test’: [‘dt’:[], ‘gt’:[]]}

Where ‘dt’ is the list of image paths and ‘gt’ is the list of texts (ground truth).

Then, just pass `--source=custom_dataset` as a parameter in the steps

--

--

No responses yet