Merge pull request meijieru#45 from ZhangXinNan/zxdev

add .gitignore, add "if torch.cuda.is_available():" in demo.py

Author: meijieru

.gitignore

@@ -0,0 +1,7 @@
+.DS_Store
+*.pth
+*.pyc
+*.pyo
+*.log
+*.tmp
+

demo.py

@@ -11,15 +11,19 @@
 img_path = './data/demo.png'
 alphabet = '0123456789abcdefghijklmnopqrstuvwxyz'
 
-model = crnn.CRNN(32, 1, 37, 256).cuda()
+model = crnn.CRNN(32, 1, 37, 256)
+if torch.cuda.is_available():
+    model = model.cuda()
 print('loading pretrained model from %s' % model_path)
 model.load_state_dict(torch.load(model_path))
 
 converter = utils.strLabelConverter(alphabet)
 
 transformer = dataset.resizeNormalize((100, 32))
 image = Image.open(img_path).convert('L')
-image = transformer(image).cuda()
+image = transformer(image)
+if torch.cuda.is_available():
+    image = image.cuda()
 image = image.view(1, *image.size())
 image = Variable(image)