-
Notifications
You must be signed in to change notification settings - Fork 52
/
evaluate.sh
executable file
·83 lines (60 loc) · 2.45 KB
/
evaluate.sh
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
#!/usr/bin/env bash
# evaluate GMFlow without refinement
# evaluate chairs & things trained model on things and sintel (Table 3 of GMFlow paper)
# the output should be:
# Number of validation image pairs: 1024
# Validation Things test set (things_clean) EPE: 3.475
# Validation Things test (things_clean) s0_10: 0.666, s10_40: 1.310, s40+: 8.968
# Number of validation image pairs: 1041
# Validation Sintel (clean) EPE: 1.495, 1px: 0.161, 3px: 0.059, 5px: 0.040
# Validation Sintel (clean) s0_10: 0.457, s10_40: 1.770, s40+: 8.257
# Number of validation image pairs: 1041
# Validation Sintel (final) EPE: 2.955, 1px: 0.209, 3px: 0.098, 5px: 0.071
# Validation Sintel (final) s0_10: 0.725, s10_40: 3.446, s40+: 17.701
CUDA_VISIBLE_DEVICES=0 python main.py \
--eval \
--resume pretrained/gmflow_things-e9887eda.pth \
--val_dataset things sintel \
--with_speed_metric
# evaluate GMFlow with refinement
# evaluate chairs & things trained model on things and sintel (Table 3 of GMFlow paper)
# the output should be:
# Validation Things test set (things_clean) EPE: 2.804
# Validation Things test (things_clean) s0_10: 0.527, s10_40: 1.009, s40+: 7.314
# Number of validation image pairs: 1041
# Validation Sintel (clean) EPE: 1.084, 1px: 0.092, 3px: 0.040, 5px: 0.028
# Validation Sintel (clean) s0_10: 0.303, s10_40: 1.252, s40+: 6.261
# Number of validation image pairs: 1041
# Validation Sintel (final) EPE: 2.475, 1px: 0.147, 3px: 0.077, 5px: 0.058
# Validation Sintel (final) s0_10: 0.511, s10_40: 2.810, s40+: 15.669
CUDA_VISIBLE_DEVICES=0 python main.py \
--eval \
--resume pretrained/gmflow_with_refine_things-36579974.pth \
--val_dataset things sintel \
--with_speed_metric \
--padding_factor 32 \
--upsample_factor 4 \
--num_scales 2 \
--attn_splits_list 2 8 \
--corr_radius_list -1 4 \
--prop_radius_list -1 1
# evaluate matched & matched on sintel
# evaluate GMFlow without refinement
CUDA_VISIBLE_DEVICES=0 python main.py \
--eval \
--evaluate_matched_unmatched \
--resume pretrained/gmflow_things-e9887eda.pth \
--val_dataset sintel
# evaluate GMFlow with refinement
CUDA_VISIBLE_DEVICES=0 python main.py \
--eval \
--evaluate_matched_unmatched \
--resume pretrained/gmflow_with_refine_things-36579974.pth \
--val_dataset sintel \
--with_speed_metric \
--padding_factor 32 \
--upsample_factor 4 \
--num_scales 2 \
--attn_splits_list 2 8 \
--corr_radius_list -1 4 \
--prop_radius_list -1 1