Spaces:
Sleeping
Sleeping
File size: 297 Bytes
899c526 |
1 2 3 4 5 6 7 8 |
import torch
import cuda_ba
neighbors = cuda_ba.neighbors
reproject = cuda_ba.reproject
def BA(poses, patches, intrinsics, target, weight, lmbda, ii, jj, kk, t0, t1, iterations=2):
return cuda_ba.forward(poses.data, patches, intrinsics, target, weight, lmbda, ii, jj, kk, t0, t1, iterations) |