(self, depth_stage_config, concat_keys=("midas_in",), *args, **kwargs)
| 1695 | """ |
| 1696 | |
| 1697 | def __init__(self, depth_stage_config, concat_keys=("midas_in",), *args, **kwargs): |
| 1698 | super().__init__(concat_keys=concat_keys, *args, **kwargs) |
| 1699 | self.depth_model = instantiate_from_config(depth_stage_config) |
| 1700 | self.depth_stage_key = concat_keys[0] |
| 1701 | |
| 1702 | @torch.no_grad() |
| 1703 | def get_input(self, batch, k, cond_key=None, bs=None, return_first_stage_outputs=False): |
nothing calls this directly
no test coverage detected