RAFT

RAFT 模型基於 RAFT：用於光流的遞迴式全對場轉換 論文。

模型建構器

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

raft_large(*[, weights, progress])	來自 RAFT：用於光流的遞迴式全對場轉換 的 RAFT 模型。
raft_small(*[, weights, progress])	來自 RAFT：用於光流的遞迴式全對場轉換 的 RAFT「小型」模型。