(depth, focal, model, u0, v0)
| 109 | return outputs |
| 110 | |
| 111 | def refine_focal(depth, focal, model, u0, v0): |
| 112 | last_scale = 1 |
| 113 | focal_tmp = np.copy(focal) |
| 114 | for i in range(1): |
| 115 | scale = refine_focal_one_step(depth, focal_tmp, model, u0, v0) |
| 116 | focal_tmp = focal_tmp / scale.item() |
| 117 | last_scale = last_scale * scale |
| 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) |
nothing calls this directly
no test coverage detected