fix image w<8 error

sbd021 · Jun 26, 2020 · 26a5826 · 26a5826
1 parent c82915d
commit 26a5826
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/crnn/network_torch.py b/crnn/network_torch.py
@@ -129,6 +129,8 @@ def predict(self,image):
 
         image       = image.view(1,1, *image.size())
         image       = Variable(image)
+        if image.size()[-1]<8:
+            return ''
         preds       = self(image)
         _, preds    = preds.max(2)
         preds       = preds.transpose(1, 0).contiguous().view(-1)
@@ -197,4 +199,4 @@ def predict_batch(self,boxes,batch_size=1):
 
 
 
-
+