RAFT

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

模型構建器

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

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