davrot/ca_imaging_funhouse
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-12 14:04:09 +02:00 committed by GitHub
parent b37a79f487
commit 8cb973f7b5
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 48 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

24
4_show.py Normal file
View file

@ -0,0 +1,24 @@
import numpy as np
import torch
from Anime import Anime
# Convert from avi to npy
filename: str = "example_data_crop"
torch_device: torch.device = torch.device(
"cuda:0" if torch.cuda.is_available() else "cpu"
)
print("Load data")
input = np.load(filename + str("_decorrelated.npy"))
data = torch.tensor(input, device=torch_device)
del input
print("loading done")
data = data.nan_to_num(nan=0.0)
#data -= data.min(dim=0, keepdim=True)[0]
ani = Anime()
ani.show(data, vmin=0.0)

24
4b_show.py Normal file
View file

@ -0,0 +1,24 @@
import numpy as np
import torch
from Anime import Anime
# Convert from avi to npy
filename: str = "example_data_crop"
torch_device: torch.device = torch.device(
"cuda:0" if torch.cuda.is_available() else "cpu"
)
print("Load data")
input = np.load(filename + str("_decorrelated.npy"))
data = torch.tensor(input, device=torch_device)
del input
print("loading done")
data = data.nan_to_num(nan=0.0)
data -= data.min(dim=0, keepdim=True)[0]
data *= data.std(dim=0, keepdim=True)
ani = Anime()
ani.show(data)