debugging trial

MPI-Dortmund · Feb 7, 2025 · d1fa2cc · d1fa2cc
1 parent ff6eeaf
commit d1fa2cc
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 2 deletions.
diff --git a/tomotwin/embed_main.py b/tomotwin/embed_main.py
@@ -273,6 +273,7 @@ def _main_():
     ui.run()
     check_for_updates()
     config = ui.get_embed_configuration()
+    print(config.padding)
     start(config)
 
 

diff --git a/tomotwin/modules/inference/argparse_embed_ui.py b/tomotwin/modules/inference/argparse_embed_ui.py
@@ -58,7 +58,6 @@ def run(self, args=None) -> None:
             self.padding = args.padding
             if args.distribution_mode == 0:
                 self.distr_mode = DistrMode.DP
-            self.padding = args.padding
 
     def get_embed_configuration(self) -> EmbedConfiguration:
         conf = EmbedConfiguration(

diff --git a/tomotwin/modules/tools/embedding_mask.py b/tomotwin/modules/tools/embedding_mask.py
@@ -153,7 +153,7 @@ def median_mode(self,
                     batch_size: int,
                     threshold: float,
                     dilation: float,
-                    padding: bool = True
+                    padding: bool 
                     ) -> np.array:
         '''
         Calculates a mask based on median embedding
@@ -162,6 +162,7 @@ def median_mode(self,
             # Embed
             emb_out_pth = os.path.join(tmp_pth, "embed")
 
+            print (padding)
             conf = EmbedConfiguration(
                 model_path=model_pth,
                 volumes_path=tomo_pth,