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:
Clemens Neudecker 2025-04-17 15:43:28 +02:00 committed by GitHub
commit d57de478eb
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 7 additions and 6 deletions

View file

@ -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: