update code

line_recognition.sh

@@ -1,9 +0,0 @@
-#!/bin/bash
-
-export CUDA_VISIBLE_DEVICES=0
-python -u ./translate_line.py -k 10 ./pretrain/WAP_params.pkl \
-	./pretrain/kindai_voc.txt \
-	./data_kindai/val.pkl \
-	./data_kindai/caption_val.txt \
-	./result/test_decode_result.txt \
-	./result/test.wer

requirements.txt

@@ -1,6 +1,7 @@
-torch==0.4.1.post2
+torch==1.4.0
 torchvision==0.2.1
 opencv-python==3.4.2.17
 scikit-image==0.14.2
 scipy==1.1.0
 Polygon3
+

utils.py

@@ -5,12 +5,6 @@ import sys
 import pickle as pkl
 import torch
 from torch import nn
-from Augmentor.Operations import Distort, Skew, RandomBrightness
-from PIL import Image
-
-d = Distort(probability=0.2, grid_height=15, grid_width=15, magnitude=10)
-sk = Skew(probability=1.0, skew_type='TILT_LEFT_RIGHT', magnitude=0.15)
-br = RandomBrightness(probability = 1.0, min_factor = 0.9, max_factor = 1.1)
 
 # load data
 def dataIterator(feature_file, label_file, dictionary, batch_size, batch_Imagesize, maxlen, maxImagesize):