qurator-spk/sbb_pixelwise_segmentation
1
0
Fork 0
mirror of https://github.com/qurator-spk/sbb_pixelwise_segmentation.git synced 2025-06-09 20:00:05 +02:00
Code Issues Releases Wiki Activity

update inference

Browse source
This commit is contained in:
vahidrezanezhad 2024-06-21 23:42:25 +02:00
parent 033cf6734b
commit c0faecec2c
1 changed files with 4 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

4
inference.py
View file

@ -557,6 +557,10 @@ class sbb_predict:
res=self.predict() res=self.predict()
if (self.task == 'classification' or self.task == 'reading_order'): if (self.task == 'classification' or self.task == 'reading_order'):
pass pass
elif self.task == 'enhancement':
if self.save:
print(self.save)
cv2.imwrite(self.save,res)
else: else:
img_seg_overlayed = self.visualize_model_output(res, self.img_org, self.task) img_seg_overlayed = self.visualize_model_output(res, self.img_org, self.task)
if self.save: if self.save: