Skip to content
Snippets Groups Projects
Commit 477833d1 authored by Andrey Aristov's avatar Andrey Aristov
Browse files

make saving tif optional

parent 7512af6c
No related branches found
Tags 0.3.2
No related merge requests found
Pipeline #82835 passed
......@@ -44,7 +44,6 @@ def align_stack(data_or_path, template16, mask2, plot=False, path_to_save=None,
print(path, stack.shape)
else:
assert data_or_path.ndim == 3 and data_or_path.shape[0] == 2
assert path_to_save is not None, "Please provide the path to save the data"
stack = data_or_path
print(f'Aligned stack will be saved to {path_to_save}')
......@@ -89,7 +88,8 @@ def align_stack(data_or_path, template16, mask2, plot=False, path_to_save=None,
aligned_stack = np.stack((aligned_bf, aligned_tritc, mask2)).astype('uint16')
imwrite(path_to_save, aligned_stack, imagej=True, metadata=META_ALIGNED)
if path_to_save is not None:
imwrite(path_to_save, aligned_stack, imagej=True, metadata=META_ALIGNED)
print(f'Saved aligned stack {path_to_save}')
return aligned_stack, tvec
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment