Output of the monodepth model.
| 150 | |
| 151 | |
| 152 | class MonodepthOutput(NamedTuple): |
| 153 | """Output of the monodepth model.""" |
| 154 | |
| 155 | # Disparity output from the monodepth model. |
| 156 | disparity: torch.Tensor |
| 157 | # Multi-level features from monodepth encoder. |
| 158 | encoder_features: list[torch.Tensor] |
| 159 | # Single-level feature from monodepth decoder. |
| 160 | decoder_features: torch.Tensor |
| 161 | # List of monodepth features to be used in gaussian predictor. |
| 162 | output_features: list[torch.Tensor] |
| 163 | # List of intermediate encoder features to be used in distillation. |
| 164 | intermediate_features: list[torch.Tensor] = [] |
| 165 | |
| 166 | |
| 167 | class MonodepthWithEncodingAdaptor(nn.Module): |