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