diff --git a/ppocr/data/lmdb_dataset.py b/ppocr/data/lmdb_dataset.py index 0c1ea9465c4b61f4cb7106160e31c217a959b3ea..2b1ccaddcea437acb3901d6b0391dd0c7b2954b7 100644 --- a/ppocr/data/lmdb_dataset.py +++ b/ppocr/data/lmdb_dataset.py @@ -37,6 +37,8 @@ class LMDBDataSet(Dataset): if self.do_shuffle: np.random.shuffle(self.data_idx_order_list) 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]) self.need_reset = True in [x < 1 for x in ratio_list]