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

sys.exit not necessary

Browse source
This commit is contained in:
Konstantin Baierer 2021-02-23 15:39:42 +01:00
parent 461d3e31cf
commit b3dd6685e7
1 changed files with 1 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

7
sbb_newspapers_org_image/eynollah.py
View file

@ -379,7 +379,6 @@ class eynollah:
K.clear_session() K.clear_session()
gc.collect() gc.collect()
# sys.exit()
img_new, num_column_is_classified = self.calculate_width_height_by_columns(img, num_col, width_early, label_p_pred) img_new, num_column_is_classified = self.calculate_width_height_by_columns(img, num_col, width_early, label_p_pred)
if img_new.shape[1] > img.shape[1]: if img_new.shape[1] > img.shape[1]:
@ -1585,7 +1584,6 @@ class eynollah:
#plt.imshow(prediction_regions_org_y[:,:,0]) #plt.imshow(prediction_regions_org_y[:,:,0])
#plt.show() #plt.show()
#sys.exit()
prediction_regions_org_y=prediction_regions_org_y[:,:,0] prediction_regions_org_y=prediction_regions_org_y[:,:,0]
mask_zeros_y=(prediction_regions_org_y[:,:]==0)*1 mask_zeros_y=(prediction_regions_org_y[:,:]==0)*1
if is_image_enhanced: if is_image_enhanced:
@ -1611,7 +1609,6 @@ class eynollah:
##plt.imshow(prediction_regions_org[:,:,0]) ##plt.imshow(prediction_regions_org[:,:,0])
##plt.show() ##plt.show()
##sys.exit()
prediction_regions_org=prediction_regions_org[:,:,0] prediction_regions_org=prediction_regions_org[:,:,0]
prediction_regions_org[(prediction_regions_org[:,:]==1) & (mask_zeros_y[:,:]==1)]=0 prediction_regions_org[(prediction_regions_org[:,:]==1) & (mask_zeros_y[:,:]==1)]=0
@ -1647,7 +1644,6 @@ class eynollah:
#plt.imshow(prediction_regions_org2[:,:,0]) #plt.imshow(prediction_regions_org2[:,:,0])
#plt.show() #plt.show()
#sys.exit()
##prediction_regions_org=prediction_regions_org[:,:,0] ##prediction_regions_org=prediction_regions_org[:,:,0]
session_region.close() session_region.close()
@ -2277,11 +2273,10 @@ class eynollah:
self.run_graphics_and_columns(text_regions_p_1, num_col_classifier, num_column_is_classified) self.run_graphics_and_columns(text_regions_p_1, num_col_classifier, num_column_is_classified)
self.logger.info("Graphics detection took %ss ", str(time.time() - t1)) self.logger.info("Graphics detection took %ss ", str(time.time() - t1))
if num_col is None: if not num_col:
self.logger.info("No columns detected, outputting an empty PAGE-XML") self.logger.info("No columns detected, outputting an empty PAGE-XML")
self.write_into_page_xml([], page_coord, self.dir_out, [], [], [], [], [], [], [], [], self.curved_line, [], []) self.write_into_page_xml([], page_coord, self.dir_out, [], [], [], [], [], [], [], [], self.curved_line, [], [])
self.logger.info("Job done in %ss", str(time.time() - t1)) self.logger.info("Job done in %ss", str(time.time() - t1))
sys.exit()
return return
t1 = time.time() t1 = time.time()