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 论文。