Number of projections added as parameter

flatironinstitute · Dec 2, 2024 · 01bd7c7 · 01bd7c7
1 parent d886701
commit 01bd7c7
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/cryo_challenge/_map_to_map/map_to_map_distance.py b/src/cryo_challenge/_map_to_map/map_to_map_distance.py
@@ -415,6 +415,7 @@ def get_distance_matrix(self, maps1, maps2, global_store_of_running_results):
         gpuID = self.config["analysis"]["zernike3d_extra_params"]["gpuID"]
         outputPath = self.config["analysis"]["zernike3d_extra_params"]["tmpDir"]
         thr = self.config["analysis"]["zernike3d_extra_params"]["thr"]
+        numProjections = self.config["analysis"]["zernike3d_extra_params"]["numProjections"]
 
         # Create output directory
         if not os.path.isdir(outputPath):
@@ -451,7 +452,7 @@ def get_distance_matrix(self, maps1, maps2, global_store_of_running_results):
         subprocess.check_call(f'eval "$({condabin_path} shell.bash hook)" &&'
                               f' conda activate flexutils-tensorflow && '
                               f'compute_distance_matrix_zernike3deep.py --references_file {references_path} '
-                              f'--targets_file {targets_paths} --out_path {outputPath} --gpu {gpuID} '
+                              f'--targets_file {targets_paths} --out_path {outputPath} --gpu {gpuID} --num_projections {numProjections}'
                               f'--thr {thr}', shell=True)
 
         # Read distance matrix