diff --git a/benchmark/PaddleOCR_DBNet/base/base_dataset.py b/benchmark/PaddleOCR_DBNet/base/base_dataset.py index e9f6c4de87d..c53945f28ad 100644 --- a/benchmark/PaddleOCR_DBNet/base/base_dataset.py +++ b/benchmark/PaddleOCR_DBNet/base/base_dataset.py @@ -1,3 +1,4 @@ +import ast # -*- coding: utf-8 -*- # @Time : 2019/12/4 13:12 # @Author : zhoujun @@ -40,7 +41,7 @@ def _init_pre_processes(self, pre_processes): else: args = aug["args"] if isinstance(args, dict): - cls = eval(aug["type"])(**args) + cls = ast.literal_eval(aug["type"])(**args) else: cls = eval(aug["type"])(args) self.aug.append(cls)