We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 0406c83 commit f186973Copy full SHA for f186973
torchvision/prototype/models/optical_flow/raft.py
@@ -21,6 +21,17 @@
21
22
23
class Raft_Large_Weights(WeightsEnum):
24
+ C_T_V1 = Weights(
25
+ # Chairs + Things, ported from original paper repo (raft-things.pth)
26
+ url="https://download.pytorch.org/models/raft_large_C_T_V1-22a6c225.pth",
27
+ transforms=RaftEval,
28
+ meta={
29
+ "recipe": "https://github.com/princeton-vl/RAFT",
30
+ "sintel_train_cleanpass_epe": 1.4411,
31
+ "sintel_train_finalpass_epe": 2.7894,
32
+ },
33
+ )
34
+
35
C_T_V2 = Weights(
36
# Chairs + Things
37
url="https://download.pytorch.org/models/raft_large_C_T_V2-1bb1363a.pth",
0 commit comments