From dd32d0f636b686e6b685d01255971e37fc40efbc Mon Sep 17 00:00:00 2001 From: "shaohua.zhang" Date: Sat, 25 Jul 2020 11:54:03 +0800 Subject: [PATCH] Update dataset_traversal.py --- ppocr/data/det/dataset_traversal.py | 16 ++++++++-------- 1 file changed, 8 insertions(+), 8 deletions(-) diff --git a/ppocr/data/det/dataset_traversal.py b/ppocr/data/det/dataset_traversal.py index fc2241cfb..89c776d50 100644 --- a/ppocr/data/det/dataset_traversal.py +++ b/ppocr/data/det/dataset_traversal.py @@ -38,14 +38,14 @@ class TrainReader(object): def __call__(self, process_id): with open(self.label_file_path, "rb") as fin: - label_infor_list = fin.readlines() - img_num = len(label_infor_list) - img_id_list = list(range(img_num)) - random.shuffle(img_id_list) - if sys.platform == "win32": - print("multiprocess is not fully compatible with Windows." - "num_workers will be 1.") - self.num_workers = 1 + label_infor_list = fin.readlines() + img_num = len(label_infor_list) + img_id_list = list(range(img_num)) + random.shuffle(img_id_list) + if sys.platform == "win32": + print("multiprocess is not fully compatible with Windows." + "num_workers will be 1.") + self.num_workers = 1 def sample_iter_reader(): for img_id in range(process_id, img_num, self.num_workers):