diff --git a/analyse_raw_data.py b/analyse_raw_data.py index 4360df4..b6208b6 100644 --- a/analyse_raw_data.py +++ b/analyse_raw_data.py @@ -8,18 +8,19 @@ import os def main( - path: str = "/data_1/hendrik/2023-07-17/M_Sert_Cre_41/raw", + path: str = "/data_1/hendrik/2021-06-17/M3859M/raw", experiment_id: int = 1, trial_id: int = 1, use_svd: bool = True, # i.e. use SVD - mask_threshold: float | None = 0.05, # Between 0 and 1.0 + mask_threshold: float | None = 0.0025, # Between 0 and 1.0 show_example_timeseries: bool = True, example_position_x: int = 280, example_position_y: int = 440, - movie_play: bool = False, - movie_vmin_scale: float = 0.5, - movie_vmax_scale: float = 0.5, - movie_enable_mask: bool = False, + movie_play: bool = True, + movie_vmin_scale: float | None = 0.05, + movie_vmax_scale: float | None = 0.1, + movie_enable_mask: bool = True, + movie_export: bool = False, export_results: bool = True, export_path: str = "Export", ): @@ -28,6 +29,14 @@ def main( else: print("Classic mode") + if movie_export is False: + movie_file: str | None = None + else: + if use_svd: + movie_file = f"SVD_Exp{experiment_id}_Trial{trial_id}.mp4" + else: + movie_file = f"Classic_Exp{experiment_id}_Trial{trial_id}.mp4" + if export_results: os.makedirs(export_path, exist_ok=True) @@ -126,10 +135,14 @@ def main( mask=mask, vmin_scale=movie_vmin_scale, vmax_scale=movie_vmax_scale, + movie_file=movie_file, ) else: ani.show( - result - 1.0, vmin_scale=movie_vmin_scale, vmax_scale=movie_vmax_scale + result - 1.0, + vmin_scale=movie_vmin_scale, + vmax_scale=movie_vmax_scale, + movie_file=movie_file, ) diff --git a/analyse_x_correlation.py b/analyse_x_correlation.py index 6659292..b450f6a 100644 --- a/analyse_x_correlation.py +++ b/analyse_x_correlation.py @@ -196,7 +196,7 @@ def make_a_x_correlation_plot( @torch.no_grad() def main( - path: str = "/data_1/hendrik/2023-07-17/M_Sert_Cre_41/raw", + path: str = "/data_1/hendrik/2021-06-17/M3859M/raw", use_svd: bool = True, remove_linear_trend: bool = False, experiment_id: int = 1,