Method forward

flowmap/flow/flow_predictor.py:33–37

(
        self,
        videos: Float[Tensor, "batch frame 3 height width"],
    )

    @abstractmethod
    def forward(
        self,
        videos: Float[Tensor, "batch frame 3 height width"],
    ) -> Float[Tensor, "batch frame-1 height width 2"]:
        pass

    @staticmethod
    def rescale_flow(