Shortcuts

raft_large

torchvision.models.optical_flow.raft_large(*, pretrained=False, progress=True, **kwargs)[source]

RAFT model from RAFT: Recurrent All Pairs Field Transforms for Optical Flow.

Please see the example below for a tutorial on how to use this model.

Parameters
  • pretrained (bool) –

    Whether to use weights that have been pre-trained on FlyingChairs + FlyingThings3D with two fine-tuning steps:

    • one on Sintel + FlyingThings3D

    • one on KittiFlow.

    This corresponds to the C+T+S/K strategy in the paper.

  • progress (bool) – If True, displays a progress bar of the download to stderr.

Returns

The model.

Return type

nn.Module

Examples using raft_large:

Docs

Access comprehensive developer documentation for PyTorch

View Docs

Tutorials

Get in-depth tutorials for beginners and advanced developers

View Tutorials

Resources

Find development resources and get your questions answered

View Resources