Imperfect fix for dewarp padding bug

braddockcg · Sep 3, 2015 · be2e6d1 · be2e6d1
1 parent 0dabf6f
commit be2e6d1
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/ocrolib/lineest.py b/ocrolib/lineest.py
@@ -50,8 +50,12 @@ def measure(self,line):
     def dewarp(self,img,cval=0,dtype=dtype('f')):
         assert img.shape==self.shape
         h,w = img.shape
-        padded = vstack([cval*ones((h,w)),img,cval*ones((h,w))])
-        center = self.center+h
+        # Changed pad to two times height to "fix" center
+        # exceeding pad height resulting in the error: 
+        # ValueError: setting an array element with a sequence.
+        # -braddock
+        padded = vstack([cval*ones((2*h,w)),img,cval*ones((2*h,w))])
+        center = self.center + 2 * h
         dewarped = [padded[center[i]-self.r:center[i]+self.r,i] for i in range(w)]
         dewarped = array(dewarped,dtype=dtype).T
         return dewarped