raft_small¶
- torchvision.models.optical_flow.raft_small(*, weights: Optional[Raft_Small_Weights] = None, progress=True, **kwargs) RAFT[source]¶
来自 RAFT 论文《RAFT: Recurrent All Pairs Field Transforms for Optical Flow》的“small”模型。
请参阅下面的示例，了解如何使用此模型。
- 参数：
weights (
Raft_Small_Weights, 可选) – 要使用的预训练权重。有关更多详细信息和可能的值，请参阅下面的Raft_Small_Weights。默认情况下，不使用预训练权重。progress (bool) – 如果为 True，则在 stderr 中显示下载进度条。默认值为 True。
**kwargs** – 传递给
torchvision.models.optical_flow.RAFT基类的参数。有关此类的更多详细信息，请参阅源代码。
- class torchvision.models.optical_flow.Raft_Small_Weights(value)[source]¶
上面的模型构建器接受以下值作为
weights参数。Raft_Small_Weights.DEFAULT等同于Raft_Small_Weights.C_T_V2。您也可以使用字符串，例如weights='DEFAULT'或weights='C_T_V1'。此处报告的指标如下。
epe是“端点误差”，表示预测光流与其真实值之间的距离（以像素为单位）。这是对所有图像的所有像素进行平均。per_image_epe类似，但平均方式不同：先独立计算每张图像的 epe，然后对所有图像进行平均。这对应于原始论文中的“Fl-epe”（有时写作“F1-epe”），并且仅用于 Kitti 数据集。fl-all也是 Kitti 特定的指标，由数据集作者定义，用于 Kitti 排行榜。它对应于 epe 小于 3px 或小于光流 L2 范数 5% 的像素的平均值。Raft_Small_Weights.C_T_V1:
这些权重从原始论文移植而来。它们在
FlyingChairs+FlyingThings3D上训练。epe（在 Sintel-Train-Cleanpass 上）
2.1231
epe（在 Sintel-Train-Finalpass 上）
3.279
per_image_epe（在 Kitti-Train 上）
7.6557
fl_all（在 Kitti-Train 上）
25.2801
最小尺寸
高=128，宽=128
参数数量
990162
训练方法
GFLOPS
47.66
文件大小
3.8 MB
推理变换可在
Raft_Small_Weights.C_T_V1.transforms中获得，并执行以下预处理操作：接受PIL.Image、批处理的(B, C, H, W)和单张(C, H, W)图像torch.Tensor对象。图像被重新缩放到[-1.0, 1.0]。Raft_Small_Weights.C_T_V2:
这些权重在
FlyingChairs+FlyingThings3D上从头开始训练。也可用作Raft_Small_Weights.DEFAULT。epe（在 Sintel-Train-Cleanpass 上）
1.9901
epe（在 Sintel-Train-Finalpass 上）
3.2831
per_image_epe（在 Kitti-Train 上）
7.5978
fl_all（在 Kitti-Train 上）
25.2369
最小尺寸
高=128，宽=128
参数数量
990162
训练方法
GFLOPS
47.66
文件大小
3.8 MB
推理变换可在
Raft_Small_Weights.C_T_V2.transforms中获得，并执行以下预处理操作：接受PIL.Image、批处理的(B, C, H, W)和单张(C, H, W)图像torch.Tensor对象。图像被重新缩放到[-1.0, 1.0]。
使用
raft_small的示例
