mirror of
https://github.com/qurator-spk/sbb_pixelwise_segmentation.git
synced 2025-06-09 20:00:05 +02:00
Merge pull request #18 from johnlockejrr/unifying-training-models
Deprecations in train.py and check an argument in inference.py
This commit is contained in:
commit
d57de478eb
2 changed files with 7 additions and 6 deletions
|
@ -567,6 +567,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:
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue