OrgLion-ML
/
deep-text-recognition-benchmark
mirror of https://github.com/clovaai/deep-text-recognition-benchmark.git
1
0
Fork 0
Code Issues Releases Wiki Activity

comment update

This commit is contained in:
Baek JeongHun 2019-11-04 04:32:46 +00:00
parent 1863c76b6a
commit 1ef8f74224
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
dataset.py
View File

@ -124,10 +124,13 @@ class LmdbDataset(Dataset):
self.nSamples = nSamples self.nSamples = nSamples
if self.opt.data_filtering_off: 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)] self.filtered_index_list = [index + 1 for index in range(self.nSamples)]
else: 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 = [] self.filtered_index_list = []
for index in range(self.nSamples): for index in range(self.nSamples):
index += 1 # lmdb starts with 1 index += 1 # lmdb starts with 1
@ -276,7 +279,7 @@ class AlignCollate(object):
resized_max_w = self.imgW resized_max_w = self.imgW
input_channel = 3 if images[0].mode == 'RGB' else 1 input_channel = 3 if images[0].mode == 'RGB' else 1
transform = NormalizePAD((input_channel, self.imgH, resized_max_w)) transform = NormalizePAD((input_channel, self.imgH, resized_max_w))
resized_images = [] resized_images = []
for image in images: for image in images:
w, h = image.size w, h = image.size