| 23 | |
| 24 | @dataclass |
| 25 | class ModelOutput: |
| 26 | depths: Float[Tensor, "batch frame height width"] |
| 27 | surfaces: Float[Tensor, "batch frame height width xyz=3"] |
| 28 | intrinsics: Float[Tensor, "batch frame 3 3"] |
| 29 | extrinsics: Float[Tensor, "batch frame 4 4"] |
| 30 | backward_correspondence_weights: Float[Tensor, "batch frame-1 height width"] |
| 31 | |
| 32 | |
| 33 | @dataclass |