Method get_camera_static

flow_poke/data.py:152–156

(self, sample: dict[str, torch.Tensor])

            return {"valid": False}

    def get_camera_static(self, sample: dict[str, torch.Tensor]) -> Bool[torch.Tensor, ""]:
        flow: Float[torch.Tensor, "l 2"] = sample["flow"]
        return (
            flow.norm(dim=-1) < self.static_camera_flow_mag_threshold
        ).float().mean() > self.static_camera_fraction_threshold

    def build_targets(self, sample: dict[str, torch.Tensor]) -> dict[str, torch.Tensor | bool]:
        try: