-
Notifications
You must be signed in to change notification settings - Fork 104
/
test_all.sh
147 lines (128 loc) · 5.59 KB
/
test_all.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
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
#!/usr/bin/env bash
DEVICE=0
echo ""
echo "-------------------------------------------------"
echo "| Test Xception on FFc23 |"
echo "-------------------------------------------------"
# put your DFDC source directory path for the extracted faces and Dataframe and uncomment the following lines
# DFDC_FACES_DIR=/your/dfdc/faces/directory
# DFDC_FACES_DF=/your/dfdc/faces/dataframe/path
# put your FF++ source directory path for the extracted faces and Dataframe and uncomment the following lines
# FFPP_FACES_DIR=/your/dfdc/faces/directory
# FFPP_FACES_DF=/your/dfdc/faces/dataframe/path
python test_model.py \
--model_path weights/binclass/net-Xception_traindb-ff-c23-720-140-140_face-scale_size-224_seed-41/bestval.pth \
--testsets ff-c23-720-140-140 dfdc-35-5-10 \
--dfdc_faces_df_path $DFDC_FACES_DF \
--dfdc_faces_dir $DFDC_FACES_DIR \
--ffpp_faces_df_path $FFPP_FACES_DF \
--ffpp_faces_dir $FFPP_FACES_DIR \
--device $DEVICE
echo ""
echo "-------------------------------------------------"
echo "| Test Xception on DFDC |"
echo "-------------------------------------------------"
python test_model.py \
--model_path weights/binclass/net-Xception_traindb-dfdc-35-5-10_face-scale_size-224_seed-41/bestval.pth \
--testsets ff-c23-720-140-140 dfdc-35-5-10 \
--dfdc_faces_df_path $DFDC_FACES_DF \
--dfdc_faces_dir $DFDC_FACES_DIR \
--ffpp_faces_df_path $FFPP_FACES_DF \
--ffpp_faces_dir $FFPP_FACES_DIR \
--device $DEVICE
echo ""
echo "-------------------------------------------------"
echo "| Test EfficientNetB4 on FFc23 |"
echo "-------------------------------------------------"
python test_model.py \
--model_path weights/binclass/net-EfficientNetB4_traindb-ff-c23-720-140-140_face-scale_size-224_seed-41/bestval.pth \
--testsets ff-c23-720-140-140 dfdc-35-5-10 \
--dfdc_faces_df_path $DFDC_FACES_DF \
--dfdc_faces_dir $DFDC_FACES_DIR \
--ffpp_faces_df_path $FFPP_FACES_DF \
--ffpp_faces_dir $FFPP_FACES_DIR \
--device $DEVICE
echo ""
echo "-------------------------------------------------"
echo "| Test EfficientNetB4 on DFDC |"
echo "-------------------------------------------------"
python test_model.py \
--model_path weights/binclass/net-EfficientNetB4_traindb-dfdc-35-5-10_face-scale_size-224_seed-41/bestval.pth \
--testsets ff-c23-720-140-140 dfdc-35-5-10 \
--dfdc_faces_df_path $DFDC_FACES_DF \
--dfdc_faces_dir $DFDC_FACES_DIR \
--ffpp_faces_df_path $FFPP_FACES_DF \
--ffpp_faces_dir $FFPP_FACES_DIR \
--device $DEVICE
echo ""
echo "-------------------------------------------------"
echo "| Test EfficientNetB4 on FFc23 (triplet) |"
echo "-------------------------------------------------"
python test_model.py \
--model_path weights/binclass/net-EfficientNetB4ST_traindb-ff-c23-720-140-140_face-scale_size-224_seed-41/bestval.pth \
--testsets ff-c23-720-140-140 dfdc-35-5-10 \
--dfdc_faces_df_path $DFDC_FACES_DF \
--dfdc_faces_dir $DFDC_FACES_DIR \
--ffpp_faces_df_path $FFPP_FACES_DF \
--ffpp_faces_dir $FFPP_FACES_DIR \
--device $DEVICE
echo ""
echo "-------------------------------------------------"
echo "| Test EfficientNetB4 on DFDC (triplet) |"
echo "-------------------------------------------------"
python test_model.py \
--model_path weights/binclass/net-EfficientNetB4ST_traindb-dfdc-35-5-10_face-scale_size-224_seed-41/bestval.pth \
--testsets ff-c23-720-140-140 dfdc-35-5-10 \
--dfdc_faces_df_path $DFDC_FACES_DF \
--dfdc_faces_dir $DFDC_FACES_DIR \
--ffpp_faces_df_path $FFPP_FACES_DF \
--ffpp_faces_dir $FFPP_FACES_DIR \
--device $DEVICE
echo ""
echo "-------------------------------------------------"
echo "| Test EfficientNetAutoAttB4 on FFc23 |"
echo "-------------------------------------------------"
python test_model.py \
--model_path weights/binclass/net-EfficientNetAutoAttB4_traindb-ff-c23-720-140-140_face-scale_size-224_seed-41/bestval.pth \
--testsets ff-c23-720-140-140 dfdc-35-5-10 \
--dfdc_faces_df_path $DFDC_FACES_DF \
--dfdc_faces_dir $DFDC_FACES_DIR \
--ffpp_faces_df_path $FFPP_FACES_DF \
--ffpp_faces_dir $FFPP_FACES_DIR \
--device $DEVICE
echo ""
echo "-------------------------------------------------"
echo "| Test EfficientNetAutoAttB4 on DFDC |"
echo "-------------------------------------------------"
python test_model.py \
--model_path weights/binclass/net-EfficientNetAutoAttB4_traindb-dfdc-35-5-10_face-scale_size-224_seed-41/bestval.pth \
--testsets ff-c23-720-140-140 dfdc-35-5-10 \
--dfdc_faces_df_path $DFDC_FACES_DF \
--dfdc_faces_dir $DFDC_FACES_DIR \
--ffpp_faces_df_path $FFPP_FACES_DF \
--ffpp_faces_dir $FFPP_FACES_DIR \
--device $DEVICE
echo ""
echo "-------------------------------------------------"
echo "| Test EfficientNetAutoAttB4 on FFc23 (triplet) |"
echo "-------------------------------------------------"
python test_model.py \
--model_path weights/binclass/net-EfficientNetAutoAttB4ST_traindb-ff-c23-720-140-140_face-scale_size-224_seed-41/bestval.pth \
--testsets ff-c23-720-140-140 dfdc-35-5-10 \
--dfdc_faces_df_path $DFDC_FACES_DF \
--dfdc_faces_dir $DFDC_FACES_DIR \
--ffpp_faces_df_path $FFPP_FACES_DF \
--ffpp_faces_dir $FFPP_FACES_DIR \
--device $DEVICE
echo ""
echo "-------------------------------------------------"
echo "| Test EfficientNetAutoAttB4 on DFDC (triplet) |"
echo "-------------------------------------------------"
python test_model.py \
--model_path weights/binclass/net-EfficientNetAutoAttB4ST_traindb-dfdc-35-5-10_face-scale_size-224_seed-41/bestval.pth \
--testsets ff-c23-720-140-140 dfdc-35-5-10 \
--dfdc_faces_df_path $DFDC_FACES_DF \
--dfdc_faces_dir $DFDC_FACES_DIR \
--ffpp_faces_df_path $FFPP_FACES_DF \
--ffpp_faces_dir $FFPP_FACES_DIR \
--device $DEVICE