From aabff3958caabf29590eb163d6c1b9034365bc48 Mon Sep 17 00:00:00 2001 From: Sunflower7788 <263037929@qq.com> Date: Thu, 29 Aug 2024 22:27:28 +0800 Subject: [PATCH] fix setting of make border epoch (#13783) --- ppocr/data/simple_dataset.py | 26 -------------------------- 1 file changed, 26 deletions(-) diff --git a/ppocr/data/simple_dataset.py b/ppocr/data/simple_dataset.py index 0a56fab2eb..4552d23a16 100644 --- a/ppocr/data/simple_dataset.py +++ b/ppocr/data/simple_dataset.py @@ -50,36 +50,10 @@ class SimpleDataSet(Dataset): self.data_idx_order_list = list(range(len(self.data_lines))) if self.mode == "train" and self.do_shuffle: self.shuffle_data_random() - - self.set_epoch_as_seed(self.seed, dataset_config) - self.ops = create_operators(dataset_config["transforms"], global_config) self.ext_op_transform_idx = dataset_config.get("ext_op_transform_idx", 2) self.need_reset = True in [x < 1 for x in ratio_list] - def set_epoch_as_seed(self, seed, dataset_config): - if self.mode == "train": - try: - border_map_id = [ - index - for index, dictionary in enumerate(dataset_config["transforms"]) - if "MakeBorderMap" in dictionary - ][0] - shrink_map_id = [ - index - for index, dictionary in enumerate(dataset_config["transforms"]) - if "MakeShrinkMap" in dictionary - ][0] - dataset_config["transforms"][border_map_id]["MakeBorderMap"][ - "epoch" - ] = (seed if seed is not None else 0) - dataset_config["transforms"][shrink_map_id]["MakeShrinkMap"][ - "epoch" - ] = (seed if seed is not None else 0) - except Exception as E: - print(E) - return - def get_image_info_list(self, file_list, ratio_list): if isinstance(file_list, str): file_list = [file_list]