Skip to content

Commit aabff39

Browse files
authored
fix setting of make border epoch (#13783)
1 parent 759317f commit aabff39

File tree

1 file changed

+0
-26
lines changed

1 file changed

+0
-26
lines changed

ppocr/data/simple_dataset.py

-26
Original file line numberDiff line numberDiff line change
@@ -50,36 +50,10 @@ def __init__(self, config, mode, logger, seed=None):
5050
self.data_idx_order_list = list(range(len(self.data_lines)))
5151
if self.mode == "train" and self.do_shuffle:
5252
self.shuffle_data_random()
53-
54-
self.set_epoch_as_seed(self.seed, dataset_config)
55-
5653
self.ops = create_operators(dataset_config["transforms"], global_config)
5754
self.ext_op_transform_idx = dataset_config.get("ext_op_transform_idx", 2)
5855
self.need_reset = True in [x < 1 for x in ratio_list]
5956

60-
def set_epoch_as_seed(self, seed, dataset_config):
61-
if self.mode == "train":
62-
try:
63-
border_map_id = [
64-
index
65-
for index, dictionary in enumerate(dataset_config["transforms"])
66-
if "MakeBorderMap" in dictionary
67-
][0]
68-
shrink_map_id = [
69-
index
70-
for index, dictionary in enumerate(dataset_config["transforms"])
71-
if "MakeShrinkMap" in dictionary
72-
][0]
73-
dataset_config["transforms"][border_map_id]["MakeBorderMap"][
74-
"epoch"
75-
] = (seed if seed is not None else 0)
76-
dataset_config["transforms"][shrink_map_id]["MakeShrinkMap"][
77-
"epoch"
78-
] = (seed if seed is not None else 0)
79-
except Exception as E:
80-
print(E)
81-
return
82-
8357
def get_image_info_list(self, file_list, ratio_list):
8458
if isinstance(file_list, str):
8559
file_list = [file_list]

0 commit comments

Comments
 (0)