RAFT

RAFT 模型基於 RAFT: Recurrent All-Pairs Field Transforms for Optical Flow 論文。

模型構建器

以下模型構建器可用於例項化 RAFT 模型，無論是否帶有預訓練權重。所有模型構建器都依賴於 torchvision.models.optical_flow.RAFT 基類。有關此類的更多詳細資訊，請參閱 原始碼。

raft_large(*[, weights, progress])	來自 RAFT: Recurrent All Pairs Field Transforms for Optical Flow 的 RAFT 模型。
raft_small(*[, weights, progress])	來自 RAFT: Recurrent All Pairs Field Transforms for Optical Flow 的 RAFT "small" 模型。