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 2024-02-03 22:44:01 +01:00 committed by GitHub
parent 75dac3f99c
commit 854bcc76fb
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 18 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

20
reproduction_effort/functions/align_cameras.py
View file

@ -19,7 +19,14 @@ def align_cameras(
batch_size: int, batch_size: int,
fill_value: float = 0, fill_value: float = 0,
) -> tuple[ ) -> tuple[
torch.Tensor, torch.Tensor, torch.Tensor, torch.Tensor, torch.Tensor, torch.Tensor torch.Tensor,
torch.Tensor,
torch.Tensor,
torch.Tensor,
torch.Tensor,
torch.Tensor,
torch.Tensor,
torch.Tensor,
]: ]:
image_alignment = ImageAlignment(default_dtype=dtype, device=device) image_alignment = ImageAlignment(default_dtype=dtype, device=device)
@ -172,4 +179,13 @@ def align_cameras(
) )
) )
return acceptor, donor, oxygenation, volume, angle_donor_volume, tvec_donor_volume return (
acceptor,
donor,
oxygenation,
volume,
angle_donor_volume,
tvec_donor_volume,
angle_refref,
tvec_refref,
)