raft_small

torchvision.models.optical_flow.raft_small(*, pretrained=False, progress=True, **kwargs)

RAFT “small” 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.

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

Returns

The model.

Return type

nn.Module

Examples using raft_small:

Optical Flow: Predicting movement with the RAFT model