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

update inference

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

4
train/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: