diff --git a/demo.py b/demo.py index c78e83ff..d82d8f69 100644 --- a/demo.py +++ b/demo.py @@ -19,7 +19,7 @@ converter = utils.strLabelConverter(alphabet) -transformer = dataset.resizeNormalize((100, 32)) +transformer = dataset.resizeNormalize((128, 32)) image = Image.open(img_path).convert('L') image = transformer(image) if torch.cuda.is_available():