davrot/gevi
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add files via upload

Browse source
This commit is contained in:
David Rotermund 2023-07-10 14:25:15 +02:00 committed by GitHub
parent b9469cd6e9
commit 0658c77467
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
test_average_calculate.py
View file

@ -5,7 +5,7 @@ from tqdm import trange
# path: str = "/data_1/robert/2021-05-05/M3852M/raw" # path: str = "/data_1/robert/2021-05-05/M3852M/raw"
path: str = "/data_1/robert/2021-05-21/M3852M/raw" path: str = "/data_1/robert/2021-05-21/M3852M/raw"
initital_mask_name: str | None = "mask.npy" initital_mask_name: str | None = None
initital_mask_update: bool = True initital_mask_update: bool = True
initital_mask_roi: bool = False # default: True initital_mask_roi: bool = False # default: True
@ -15,7 +15,7 @@ start_position: int = 0
start_position_coefficients: int = 100 start_position_coefficients: int = 100
remove_heartbeat: bool = True # i.e. use SVD remove_heartbeat: bool = True # i.e. use SVD
bin_size: int = 4 bin_size: int = 4
threshold: float | None = 0.05 # Between 0 and 1.0 threshold: float | None = None # Between 0 and 1.0
display_logging_messages: bool = False display_logging_messages: bool = False
@ -75,8 +75,11 @@ for trial_id in trange(0, len(list_of_trials)):
result += result_temp result += result_temp
assert result is not None assert result is not None
assert mask is not None
result /= n result /= n
np.savez("result.npz", result=result.cpu(), mask=mask.cpu()) np.save("result.npy", result.cpu())
if mask is not None:
np.save("mask.npy", mask.cpu())