Extend segmentation check for uint64 as well

This solves #77.
ocropus-archive · Dec 25, 2017 · 9bcfc04 · 9bcfc04
1 parent 8cfce57
commit 9bcfc04
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ocrolib/toplevel.py b/ocrolib/toplevel.py
@@ -410,7 +410,7 @@ def PATCH(a):
 
 @makeargcheck("expected a segmentation image")
 def SEGMENTATION(a):
-    return isinstance(a,numpy.ndarray) and a.ndim==2 and a.dtype in ['int32','int64']
+    return isinstance(a,numpy.ndarray) and a.ndim==2 and a.dtype in int_dtypes
 @makeargcheck("expected a segmentation with white background")
 def WHITESEG(a):
     return numpy.amax(a)==0xffffff