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。