(self, frames, depths, cond_times=None)
| 133 | self.depth_prior = DepthAnythingWrapper(opt.depth_model_name) |
| 134 | |
| 135 | def forward(self, frames, depths, cond_times=None): |
| 136 | with torch.no_grad(): |
| 137 | predicted_depth = self.depth_prior(depths).detach() # [B, H, W] |
| 138 | predicted_depth = F.interpolate(predicted_depth[:, None], size=frames.shape[-2:], mode="bilinear", align_corners=True) |
| 139 | return predicted_depth # [B, C, H, W] |
| 140 | |
| 141 | def run_inference(opt: Options): |
| 142 | train_dataset = DAVISDataset(opt, training=True) |
nothing calls this directly
no outgoing calls
no test coverage detected