From 603d463e83aad21cff1b27bb254cafef1b07a15c Mon Sep 17 00:00:00 2001 From: Baek JeongHun Date: Mon, 4 Nov 2019 04:52:51 +0000 Subject: [PATCH] comment update --- train.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/train.py b/train.py index c30dce6eea..e24e5119bb 100755 --- a/train.py +++ b/train.py @@ -21,6 +21,11 @@ def train(opt): """ dataset preparation """ + if not opt.data_filtering_off: + print('Filtering the images containing characters which are not in opt.character') + print('Filtering the images whose label is longer than opt.batch_max_length') + # see https://github.com/clovaai/deep-text-recognition-benchmark/blob/master/dataset.py#L130 + opt.select_data = opt.select_data.split('-') opt.batch_ratio = opt.batch_ratio.split('-') train_dataset = Batch_Balanced_Dataset(opt)