From 0dfb536fa6eec201682d01c6e6dc256e33029db3 Mon Sep 17 00:00:00 2001
From: lgcy <865562216@qq.com>
Date: Tue, 17 May 2022 21:00:39 +0800
Subject: [PATCH] Update lmdb_dataset.py

---
 ppocr/data/lmdb_dataset.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/ppocr/data/lmdb_dataset.py b/ppocr/data/lmdb_dataset.py
index 0c1ea946..2b1ccadd 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]
-- 
GitLab