Skip to content

GitLab

Unverified Commit 0dfb536f authored by lgcy's avatar lgcy Committed by GitHub
Browse files

Update lmdb_dataset.py

parent 5cab4c0c
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
ppocr/data/lmdb_dataset.py
View file @ 0dfb536f
...@@ -37,6 +37,8 @@ class LMDBDataSet(Dataset): ...@@ -37,6 +37,8 @@ class LMDBDataSet(Dataset):
if self.do_shuffle: if self.do_shuffle:
np.random.shuffle(self.data_idx_order_list) np.random.shuffle(self.data_idx_order_list)
self.ops = create_operators(dataset_config['transforms'], global_config) self.ops = create_operators(dataset_config['transforms'], global_config)
self.ext_op_transform_idx = dataset_config.get("ext_op_transform_idx",
2)
ratio_list = dataset_config.get("ratio_list", [1.0]) ratio_list = dataset_config.get("ratio_list", [1.0])
self.need_reset = True in [x < 1 for x in ratio_list] self.need_reset = True in [x < 1 for x in ratio_list]
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment