RAFT¶

get_raft_optical_flow(
frame_data: FrameData,
ref_data: List[FrameData],
) → List[Tensor][source]¶

Apply a RAFT model on its frame and each of its reference. Return the raft-estimated optical flows.

Parameters:

  • frame_data (FrameData) – Data for the frame to be encoded.

  • ref_data (List[FrameData]) – Data of the references.

Returns:

List of optical flows. Each of them with a shape of [1, 2, H, W].

Return type:

List[Tensor]