Skip to content

Commit

Permalink
comment update
Browse files Browse the repository at this point in the history
  • Loading branch information
Baek JeongHun committed Nov 4, 2019
1 parent 1863c76 commit 1ef8f74
Showing 1 changed file with 6 additions and 3 deletions.
9 changes: 6 additions & 3 deletions dataset.py
Original file line number Diff line number Diff line change
Expand Up @@ -124,10 +124,13 @@ def __init__(self, root, opt):
self.nSamples = nSamples

if self.opt.data_filtering_off:
# for fast check with no filtering
# for fast check or benchmark evaluation with no filtering
self.filtered_index_list = [index + 1 for index in range(self.nSamples)]
else:
# Filtering
""" Filtering part
If you want to evaluate IC152077 & CUTE80 datasets which have special character labels, use --data_filtering_off and evaluation with this snippet (only evaluate on alphabets and digits).
https://github.com/clovaai/deep-text-recognition-benchmark/blob/master/dataset.py#L186-L188
"""
self.filtered_index_list = []
for index in range(self.nSamples):
index += 1 # lmdb starts with 1
Expand Down Expand Up @@ -276,7 +279,7 @@ def __call__(self, batch):
resized_max_w = self.imgW
input_channel = 3 if images[0].mode == 'RGB' else 1
transform = NormalizePAD((input_channel, self.imgH, resized_max_w))

resized_images = []
for image in images:
w, h = image.size
Expand Down

0 comments on commit 1ef8f74

Please sign in to comment.