fixed module path

Author: CarlaRodriguez96

experiments/four_f_optimizer.py

@@ -1,9 +1,19 @@
+import os
+import sys
+
+# Setting the path for XLuminA modules:
+current_path = os.path.abspath(os.path.join('..'))
+module_path = os.path.join(current_path)
+
+if module_path not in sys.path:
+    sys.path.append(module_path)
+
 from four_f_optical_table import *
-# from xlumina.toolbox import MultiHDF5DataLoader
+from xlumina.toolbox import MultiHDF5DataLoader
 import time
 import jax
-from jax import grad, jit
-from jax.example_libraries import optimizers
+import optax
+from jax import jit
 import numpy as np
 import jax.numpy as jnp
 
@@ -82,7 +92,7 @@ def update(params, opt_state, input_fields, target_fields):
 
 # Optimizer settings
 num_iterations = 50000
-num_samples = 50
+num_samples = 1
 # Step size engineering:
 STEP_SIZE = 0.01
 WEIGHT_DECAY = 0.0001

experiments/generate_synthetic_data.py

@@ -1,17 +1,19 @@
 # Setting the path for XLuminA modules:
 import os
 import sys
+
+# Setting the path for XLuminA modules:
 current_path = os.path.abspath(os.path.join('..'))
-dir_path = os.path.dirname(current_path)
-module_path = os.path.join(dir_path)
+module_path = os.path.join(current_path)
+
 if module_path not in sys.path:
     sys.path.append(module_path)
-
+    
 import numpy as np
 import jax.numpy as jnp
-from __init__ import um, nm, mm, degrees, radians
+from __init__ import um, nm
 from xlumina.wave_optics import *
-from wave_optics import *
+from xlumina.optical_elements import *
 from xlumina.toolbox import space
 import h5py
 
@@ -95,7 +97,7 @@ def generate_synthetic_annular(gb, num_samples):
     input_fields = jnp.vstack([input_circles, input_squares, input_annular]) 
     target_fields = jnp.vstack([target_circles, target_squares, target_annular]) 
 
-    filename = f"training_data_4f/synthetic_data_{s}.hdf5"
+    filename = f"new_training_data_4f/new_synthetic_data_{s+150}.hdf5"
 
     with h5py.File(filename, 'w') as hdf:
         # Create datasets for your data

experiments/hybrid_optimizer.py

@@ -1,8 +1,10 @@
 import os
 import sys
+
+# Setting the path for XLuminA modules:
 current_path = os.path.abspath(os.path.join('..'))
-dir_path = os.path.dirname(current_path)
-module_path = os.path.join(dir_path)
+module_path = os.path.join(current_path)
+
 if module_path not in sys.path:
     sys.path.append(module_path)
 

experiments/hybrid_sharp_optical_table.py

@@ -1,9 +1,11 @@
 # Setting the path for XLuminA modules:
 import os
 import sys
+
+# Setting the path for XLuminA modules:
 current_path = os.path.abspath(os.path.join('..'))
-dir_path = os.path.dirname(current_path)
-module_path = os.path.join(dir_path)
+module_path = os.path.join(current_path)
+
 if module_path not in sys.path:
     sys.path.append(module_path)
 

experiments/hybrid_sted_optical_table.py

@@ -1,9 +1,11 @@
 # Setting the path for XLuminA modules:
 import os
 import sys
+
+# Setting the path for XLuminA modules:
 current_path = os.path.abspath(os.path.join('..'))
-dir_path = os.path.dirname(current_path)
-module_path = os.path.join(dir_path)
+module_path = os.path.join(current_path)
+
 if module_path not in sys.path:
     sys.path.append(module_path)
 

experiments/hybrid_with_fixed_PM.py

@@ -1,13 +1,15 @@
 # Setting the path for XLuminA modules:
 import os
 import sys
+
+# Setting the path for XLuminA modules:
 current_path = os.path.abspath(os.path.join('..'))
-dir_path = os.path.dirname(current_path)
-module_path = os.path.join(dir_path)
+module_path = os.path.join(current_path)
+
 if module_path not in sys.path:
     sys.path.append(module_path)
-
-from xlumina.__init__ import um, nm, cm, mm
+    
+from xlumina.__init__ import um, nm, mm
 from xlumina.vectorized_optics import *
 from xlumina.optical_elements import hybrid_setup_fixed_slms_fluorophores, hybrid_setup_fixed_slms
 from xlumina.loss_functions import vectorized_loss_hybrid

experiments/six_times_six_ansatz_with_fixed_PM.py

@@ -1,9 +1,11 @@
 # Setting the path for XLuminA modules:
 import os
 import sys
+
+# Setting the path for XLuminA modules:
 current_path = os.path.abspath(os.path.join('..'))
-dir_path = os.path.dirname(current_path)
-module_path = os.path.join(dir_path)
+module_path = os.path.join(current_path)
+
 if module_path not in sys.path:
     sys.path.append(module_path)