mirror of
https://github.com/qurator-spk/sbb_pixelwise_segmentation.git
synced 2025-06-09 20:00:05 +02:00
update inference
This commit is contained in:
parent
033cf6734b
commit
c0faecec2c
1 changed files with 4 additions and 0 deletions
|
@ -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:
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue