From 1cbe4bf26cd35e334662215c56b291e23f38ab51 Mon Sep 17 00:00:00 2001 From: WenmuZhou <572459439@qq.com> Date: Fri, 14 Jan 2022 04:41:03 +0000 Subject: [PATCH] add todo --- ppocr/data/collate_fn.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/ppocr/data/collate_fn.py b/ppocr/data/collate_fn.py index 13aa2a6f..89c6b4fd 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: -- GitLab