From a44c1ed24663e27d38b079bcfa1984073357ecd5 Mon Sep 17 00:00:00 2001 From: wenlihaoyu Date: Fri, 8 Dec 2017 10:58:44 +0800 Subject: [PATCH] =?UTF-8?q?=E6=96=B0=E5=A2=9Eocr=E8=AE=AD=E7=BB=83?= =?UTF-8?q?=E4=BB=A3=E7=A0=81?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- train/keras-train/trainbatch.py | 2 +- train/pytorch-train/crnn_main.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/train/keras-train/trainbatch.py b/train/keras-train/trainbatch.py index 604db66..cb86860 100644 --- a/train/keras-train/trainbatch.py +++ b/train/keras-train/trainbatch.py @@ -15,7 +15,7 @@ train_loader = torch.utils.data.DataLoader( testSize = 64 test_loader = torch.utils.data.DataLoader( - train_dataset, batch_size=testSize, + test_dataset, batch_size=testSize, shuffle=True, sampler=sampler, num_workers=int(workers), collate_fn=dataset.alignCollate(imgH=imgH, imgW=imgW, keep_ratio=keep_ratio)) diff --git a/train/pytorch-train/crnn_main.py b/train/pytorch-train/crnn_main.py index fb6efc4..d40e1b5 100644 --- a/train/pytorch-train/crnn_main.py +++ b/train/pytorch-train/crnn_main.py @@ -20,8 +20,8 @@ import models.crnn as crnn #alphabet += u''.join(list(set(newChars) - set(alphabet))) parser = argparse.ArgumentParser() -parser.add_argument('--trainroot', help='path to dataset',default='../trainval/data') -parser.add_argument('--valroot', help='path to dataset',default='../trainval/data') +parser.add_argument('--trainroot', help='path to dataset',default='../data/lmdb/train') +parser.add_argument('--valroot', help='path to dataset',default='../data/lmdb/val') parser.add_argument('--workers', type=int, help='number of data loading workers', default=4) parser.add_argument('--batchSize', type=int, default=128, help='input batch size') parser.add_argument('--imgH', type=int, default=32, help='the height of the input image to network') -- GitLab