Skip to content

Commit f186973

Browse files
committed
Add pretrained weights from paper's repo as V1
1 parent 0406c83 commit f186973

File tree

1 file changed

+11
-0
lines changed
  • torchvision/prototype/models/optical_flow

1 file changed

+11
-0
lines changed

torchvision/prototype/models/optical_flow/raft.py

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -21,6 +21,17 @@
2121

2222

2323
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+
2435
C_T_V2 = Weights(
2536
# Chairs + Things
2637
url="https://download.pytorch.org/models/raft_large_C_T_V2-1bb1363a.pth",

0 commit comments

Comments
 (0)