File tree Expand file tree Collapse file tree 2 files changed +3
-3
lines changed
torchvision/prototype/models/optical_flow Expand file tree Collapse file tree 2 files changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -62,4 +62,4 @@ You can also evaluate on Kitti train:
62
62
```
63
63
torchrun --nproc_per_node 1 --nnodes 1 train.py --val-dataset kitti --batch-size 1 --dataset-root $dataset_root --model raft_large --pretrained
64
64
Kitti val epe: 4.7968 1px: 0.6388 3px: 0.8197 5px: 0.8661 per_image_epe: 4.5118 f1: 16.0679
65
- ```
65
+ ```
Original file line number Diff line number Diff line change @@ -97,11 +97,11 @@ class Raft_Small_Weights(WeightsEnum):
97
97
)
98
98
C_T_V2 = Weights (
99
99
# Chairs + Things
100
- url = "https://github.com/ pytorch/vision/tree/main/references/optical_flow " ,
100
+ url = "https://download. pytorch.org/models/raft_small_C_T_V2-01064c6d.pth " ,
101
101
transforms = RaftEval ,
102
102
meta = {
103
103
** _COMMON_META ,
104
- "recipe" : "https://github.com/princeton-vl/RAFT " ,
104
+ "recipe" : "https://github.com/pytorch/vision/tree/main/references/optical_flow " ,
105
105
"sintel_train_cleanpass_epe" : 1.9901 ,
106
106
"sintel_train_finalpass_epe" : 3.2831 ,
107
107
"kitti_train_per_image_epe" : 7.5978 ,
You can’t perform that action at this time.
0 commit comments