From bf5837bf6e4c44add1d401a9912fd1bd599df780 Mon Sep 17 00:00:00 2001 From: vahidrezanezhad Date: Fri, 9 Aug 2024 13:20:09 +0200 Subject: [PATCH] update --- train/inference.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train/inference.py b/train/inference.py index 6054b01..8d0a572 100644 --- a/train/inference.py +++ b/train/inference.py @@ -28,7 +28,7 @@ Tool to load model and predict for given image. """ class sbb_predict: - def __init__(self,image, model, task, config_params_model, patches, save, ground_truth, xml_file, out, min_area): + def __init__(self,image, model, task, config_params_model, patches, save, save_layout, ground_truth, xml_file, out, min_area): self.image=image self.patches=patches self.save=save