|
5 | 5 | performed after the acquisition and not in "real-time".
|
6 | 6 | """
|
7 | 7 |
|
8 |
| -from spas.acquisition import init_2arms, setup_cam, AcquisitionParameters, setup_2arms, setup, acquire, acquire_2arms, snapshot, disconnect_2arms, captureVid, displaySpectro, setup_tuneSpectro, change_patterns |
9 |
| -from spas.metadata import MetaData, func_path, save_metadata_2arms |
10 |
| -from spas.reconstruction import reconstruction_hadamard |
| 8 | +from spas.acquisition_SPC2D import init_2arms, setup_cam_ids, AcquisitionParameters_SPC2D, setup_2arms, setup, acquire, acquire_2arms, snapshot, disconnect_2arms, captureVid, displaySpectro, setup_tuneSpectro, change_patterns |
| 9 | +from spas.metadata_SPC2D import MetaData_SPC2D, func_path, save_metadata_2arms |
| 10 | +from spas.reconstruction_SPC2D import reconstruction_hadamard_2D |
11 | 11 | from spas.reconstruction_nn import ReconstructionParameters, setup_reconstruction, reorder_subsample
|
12 | 12 | from spas.noise import load_noise
|
13 | 13 | from spas.visualization import snapshotVisu, displayVid, plot_reco_without_NN, plot_reco_with_NN, extract_ROI_coord
|
|
29 | 29 | camPar = captureVid(camPar)
|
30 | 30 | #%% Set Camera Parameters
|
31 | 31 | # It is advice to execute this cell twice to take into account the parameter changement
|
32 |
| -camPar = setup_cam(camPar, |
| 32 | +camPar = setup_cam_ids(camPar, |
33 | 33 | pixelClock = 474, # Allowed values : [118, 237, 474] (MHz)
|
34 | 34 | fps = 220, # FrameRate boundary : [1 - No value(depend of the AOI size)]
|
35 | 35 | Gain = 0, # Gain boundary : [0 100]
|
|
73 | 73 | if 'mask_index' not in locals(): mask_index = []; x_mask_coord = []; y_mask_coord = [] # execute "mask_index = []" to not apply the mask
|
74 | 74 |
|
75 | 75 | if all_path.aborted == False:
|
76 |
| - metadata = MetaData( |
| 76 | + metadata = MetaData_SPC2D( |
77 | 77 | output_directory = all_path.subfolder_path,
|
78 | 78 | pattern_order_source = 'C:/openspyrit/spas/stats/pattern_order_' + scan_mode + '_' + str(Np) + 'x' + str(Np) + '.npz',
|
79 | 79 | pattern_source = 'C:/openspyrit/spas/Patterns/' + scan_mode + '_' + str(Np) + 'x' + str(Np),
|
|
89 | 89 | force_change = False)
|
90 | 90 | except: pass
|
91 | 91 |
|
92 |
| - acquisition_parameters = AcquisitionParameters(pattern_compression = pattern_compression, pattern_dimension_x = Np, pattern_dimension_y = Np, |
| 92 | + acquisition_parameters = AcquisitionParameters_SPC2D(pattern_compression = pattern_compression, pattern_dimension_x = Np, pattern_dimension_y = Np, |
93 | 93 | zoom = zoom, xw_offset = xw_offset, yh_offset = yh_offset, mask_index = mask_index,
|
94 | 94 | x_mask_coord = x_mask_coord, y_mask_coord = y_mask_coord)
|
95 | 95 |
|
|
130 | 130 | save_metadata_2arms(metadata, DMD_params, spectrometer_params, camPar, acquisition_parameters)
|
131 | 131 | #%% Hadamard Reconstruction
|
132 | 132 | Q = wh.walsh2_matrix(Np)
|
133 |
| -GT = reconstruction_hadamard(acquisition_parameters, 'walsh', Q, spectral_data, Np) |
| 133 | +GT = reconstruction_hadamard_2D(acquisition_parameters, 'walsh', Q, spectral_data, Np) |
134 | 134 | plot_reco_without_NN(acquisition_parameters, GT, all_path)
|
135 | 135 | #%% Neural Network setup (executed it just one time)
|
136 | 136 | network_param = ReconstructionParameters(
|
|
0 commit comments