fix combine 2 lmdb dataset

wenwenyu / MASTER-pytorch

Code for the paper "MASTER: Multi-Aspect Non-local Network for Scene Text Recognition" (Pattern Recognition 2021)

https://arxiv.org/abs/1910.02562

MIT License

277 stars 53 forks source link

fix combine 2 lmdb dataset #5

Closed cuongngm closed 3 years ago

cuongngm commented 3 years ago

Last commit met problem of incorrectly loading image_keys and label_keys and making the model not converge. This commit edit way to combine lmdb dataset ST+ MJ using torch.utils.data.ConcatDataset

wenwenyu commented 3 years ago

Thanks!