Return dimensions of output feature maps.
(self)
| 227 | ) |
| 228 | |
| 229 | def get_feature_dims(self) -> list[int]: |
| 230 | """Return dimensions of output feature maps.""" |
| 231 | dims = [] |
| 232 | if self.return_encoder_features: |
| 233 | dims.extend(self.monodepth_predictor.encoder.dims_encoder) |
| 234 | |
| 235 | if self.return_decoder_features: |
| 236 | dims.append(self.monodepth_predictor.decoder.dim_out) |
| 237 | |
| 238 | return dims |
| 239 | |
| 240 | def internal_resolution(self) -> int: |
| 241 | """Return the internal image size of the network.""" |