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

Update inference.py to check if save_layout was passed as argument otherwise can give an cv2 error

Browse source
This commit is contained in:
johnlockejrr 2024-10-19 13:21:29 -07:00 • committed by GitHub
parent 3f354e1c34
commit a524f8b1a7
1 changed files with 1 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

1
train/inference.py
View file

@ -566,6 +566,7 @@ class sbb_predict:
img_seg_overlayed, only_layout = self.visualize_model_output(res, self.img_org, self.task)
if self.save:
cv2.imwrite(self.save,img_seg_overlayed)
if self.save_layout:
cv2.imwrite(self.save_layout, only_layout)
if self.ground_truth: