qurator-spk/eynollah
1
0
Fork 0
mirror of https://github.com/qurator-spk/eynollah.git synced 2025-07-08 02:19:57 +02:00
Code Issues Projects Releases Wiki Activity

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

Browse source
This commit is contained in:
vahidrezanezhad 2024-12-04 16:41:00 +01:00
parent 6aad006f4c
commit 871d7bfc5a
1 changed files with 4 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

3
src/eynollah/eynollah.py
View file

@ -4013,6 +4013,9 @@ class Eynollah:
inference_bs = 3 inference_bs = 3
input_1= np.zeros( (inference_bs, height1, width1,3)) input_1= np.zeros( (inference_bs, height1, width1,3))
starting_list_of_regions = [] starting_list_of_regions = []
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])) ) starting_list_of_regions.append( list(range(labels_con.shape[2])) )
index_update = 0 index_update = 0
index_selected = starting_list_of_regions[0] index_selected = starting_list_of_regions[0]