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,
fill_value: float = 0,
) -> 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)
@ -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,
)