fixed: machine based reading order cause tuple index out of range error if number of textregion is one.

3 weeks ago · 871d7bfc5a
parent 6aad006f4c
commit 871d7bfc5a
1 changed files with 4 additions and 1 deletions
--- a/src/eynollah/eynollah.py
+++ b/src/eynollah/eynollah.py
@ -4013,7 +4013,10 @@ class Eynollah:
        inference_bs = 3
        input_1= np.zeros( (inference_bs, height1, width1,3))
        starting_list_of_regions = []
-        starting_list_of_regions.append( list(range(labels_con.shape[2])) )
+        if len(co_text_all)<=1:
+            starting_list_of_regions.append( list(range(1)) )
+        else:
+            starting_list_of_regions.append( list(range(labels_con.shape[2])) )
        index_update = 0
        index_selected = starting_list_of_regions[0]
        #print(labels_con.shape[2],"number of regions for reading order")