Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Train with lmdb dataset #1

Open
CuongNN218 opened this issue Apr 13, 2021 · 5 comments
Open

Train with lmdb dataset #1

CuongNN218 opened this issue Apr 13, 2021 · 5 comments

Comments

@CuongNN218
Copy link

I want to train Master with LMDB dataset but I don't see any args related to image size. How can I train with fixed image size ?

@CuongNN218
Copy link
Author

@wenwenyu and how can I train with only one GPU

@wenwenyu
Copy link
Owner

Hi, LmdbDataset has added img_w and img_h args. Training with one GPU can use debug model. e.g. python train.py -c configs/config.json -d 1 -dist false, and -d specify which one gpu will be used.

@CuongNN218
Copy link
Author

CuongNN218 commented Apr 14, 2021

how can I create a lmdb dataset to work with it. Maybe your key in lmdb file is quite different from that of deep-text-recognition-benchmark @wenwenyu

@wenwenyu
Copy link
Owner

Yes, the key in the lmdb file is different from that of deep-text-recognition-benchmark. Two ways to deal with this difference.

  • You can modify the key to adapt your lmdb dataset, including nSamples, image_key, and label_key

  • Refer to the example of creating lmdb dataset to satisfy our LmdbDataset

@qyfff
Copy link

qyfff commented Feb 10, 2023

我有3个gpu在一个机器上怎么训练

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

3 participants