(depth_wshift, model, focal, u0, v0)
| 118 | return torch.tensor([[last_scale]]) |
| 119 | |
| 120 | def refine_shift(depth_wshift, model, focal, u0, v0): |
| 121 | depth_wshift_tmp = np.copy(depth_wshift) |
| 122 | last_shift = 0 |
| 123 | for i in range(1): |
| 124 | shift = refine_shift_one_step(depth_wshift_tmp, model, focal, u0, v0) |
| 125 | shift = shift if shift.item() < 0.7 else torch.tensor([[0.7]]) |
| 126 | depth_wshift_tmp -= shift.item() |
| 127 | last_shift += shift.item() |
| 128 | return torch.tensor([[last_shift]]) |
| 129 | |
| 130 | def reconstruct_3D(depth, f): |
| 131 | """ |
nothing calls this directly
no test coverage detected