Skip to content

GitLab

Commit 1cbe4bf2 authored by WenmuZhou's avatar WenmuZhou
Browse files

add todo

parent 23eb335d
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
ppocr/data/collate_fn.py
View file @ 1cbe4bf2
...@@ -24,6 +24,7 @@ class DictCollator(object): ...@@ -24,6 +24,7 @@ class DictCollator(object):
""" """
def __call__(self, batch): def __call__(self, batch):
# todo：support batch operators
data_dict = defaultdict(list) data_dict = defaultdict(list)
to_tensor_keys = [] to_tensor_keys = []
for sample in batch: for sample in batch:
...@@ -43,6 +44,7 @@ class ListCollator(object): ...@@ -43,6 +44,7 @@ class ListCollator(object):
""" """
def __call__(self, batch): def __call__(self, batch):
# todo：support batch operators
data_dict = defaultdict(list) data_dict = defaultdict(list)
to_tensor_idxs = [] to_tensor_idxs = []
for sample in batch: for sample in batch:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment