diff --git a/ppocr/data/collate_fn.py b/ppocr/data/collate_fn.py index 13aa2a6fc8fd095e220365533e4a6379b1e14ba5..89c6b4fd5ae151e1d703ea5c59abf0177dfc3a8b 100644 --- a/ppocr/data/collate_fn.py +++ b/ppocr/data/collate_fn.py @@ -24,6 +24,7 @@ class DictCollator(object): """ def __call__(self, batch): + # todoļ¼šsupport batch operators data_dict = defaultdict(list) to_tensor_keys = [] for sample in batch: @@ -43,6 +44,7 @@ class ListCollator(object): """ def __call__(self, batch): + # todoļ¼šsupport batch operators data_dict = defaultdict(list) to_tensor_idxs = [] for sample in batch: