config.yaml
3.36 KB
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
148
149
150
MODEL:
CLS_FIGHTING_2: '/home/jwq/models/cls_fighting_2/cls_fighting_2_v0.0.1.pth'
CLS_EMOTION: '/home/jwq/models/cls_emotion/v0.1.0.m'
FEATURE_EMOTION: '/home/jwq/models/feature_emotion/FerPlus3.h5'
CLS_AUDIO: '/home/jwq/models/cls_audio/v0.0.1.m'
CLS_CLASS: '/home/jwq/models/cls_class/v_0.0.1_xgb.pkl'
CLS_VIDEO: '/home/jwq/models/cls_video/v0.4.1.pth'
CLS_POSE: '/home/jwq/models/cls_pose/v0.0.1.pth'
CLS_FLOW: '/home/jwq/models/cls_flow/v0.1.1.pth'
CLS_BG: '/home/jwq/models/cls_bg/v0.1.1.pkl'
CLS_PERSON: '/home/jwq/models/cls_person/v0.1.1.pkl'
THRESHOLD:
FACES_THRESHOLD: 0.6
FILTER:
VIDEO:
VIDEO_DIR: '/home/jwq/Desktop/VGAF_EmotiW/Val'
LABEL_PATH: '/home/jwq/Desktop/VGAF_EmotiW/Val_labels.txt'
VIDEO_SAVE_DIR: '/home/jwq/Desktop/tmp/video'
AUDIO_SAVE_DIR: '/home/jwq/npys/'
FRAME_SAVE_DIR: '/home/jwq/Desktop/tmp/frame'
# FRAME_SAVE_DIR: '/home/jwq/Desktop/VGAF_EmotiW_class/train_frame'
FLOW_SAVE_DIR: '/home/jwq/Desktop/tmp/flow'
POSE_FRAME_SAVE_DIR: '/home/jwq/Desktop/tmp/pose_frame'
FRAME_LIST_DIR: '/home/jwq/Desktop/tmp/file_list'
IS10_FEATURE_NP_DIR: '/home/jwq/npys'
IS10_FEATURE_CSV_DIR: '/home/jwq/Desktop/tmp/is10'
# FACE_FEATURE_DIR: '/home/jwq/Desktop/tmp/face_feature_retina'
# FACE_FEATURE_DIR: '/data2/retinaface/random_face_frame_features/'
FACE_FEATURE_DIR: '/data1/segment/'
# FACE_FEATURE_DIR: '/home/jwq/npys/'
FACE_IMAGE_DIR: '/data2/retinaface/train/'
CLASS_FEATURE_DIR: '/home/jwq/Desktop/tmp/class'
PREFIX: 'img_{:05d}.jpg'
FLOW_PREFIX: 'flow_{}_{:05d}.jpg'
THREAD_NUM: 10
FPS: 5
VIDEO_FILTER:
TEST_SEGMENT: 8
TEST_CROP: 1
BATCH_SIZE: 1
INPUT_SIZE: 224
MODALITY: 'RGB'
ARCH: 'resnet50'
RESULT_FILE: 'video_filter.txt'
VIDEO_1_FILTER:
TEST_SEGMENT: 8
TEST_CROP: 1
BATCH_SIZE: 1
INPUT_SIZE: 224
MODALITY: 'RGB'
ARCH: 'resnet34'
RESULT_FILE: 'video_1_filter.txt'
EMOTION:
INTERVAL: 1
INPUT_SIZE: 224
RESULT_FILE: 'emotion_filter.txt'
EMOTION_1:
RESULT_FILE: 'emotion_1_filter.txt'
DATA_NAME: 'val.npy'
ARGUE:
DIMENSION: 1582
RESULT_FILE: 'argue_filter.txt'
FIGHTING:
TEST_SEGMENT: 8
TEST_CROP: 1
BATCH_SIZE: 1
INPUT_SIZE: 224
MODALITY: 'RGB'
ARCH: 'resnet50'
RESULT_FILE: 'fighting_filter.txt'
FIGHTING_2:
TEST_SEGMENT: 8
TEST_CROP: 1
BATCH_SIZE: 1
INPUT_SIZE: 224
MODALITY: 'RGB'
ARCH: 'resnet50'
RESULT_FILE: 'fighting_2_filter.txt'
MEETING:
TEST_SEGMENT: 8
TEST_CROP: 1
BATCH_SIZE: 1
INPUT_SIZE: 224
MODALITY: 'RGB'
ARCH: 'resnet50'
RESULT_FILE: 'meeting_filter.txt'
TROOPS:
TEST_SEGMENT: 8
TEST_CROP: 1
BATCH_SIZE: 1
INPUT_SIZE: 224
MODALITY: 'RGB'
ARCH: 'resnet50'
RESULT_FILE: 'troops_filter.txt'
FLOW:
TEST_SEGMENT: 8
TEST_CROP: 1
BATCH_SIZE: 1
INPUT_SIZE: 224
MODALITY: 'Flow'
ARCH: 'resnet50'
RESULT_FILE: 'flow_filter.txt'
FINAL:
RESULT_FILE: 'final.txt'
ERROR_FILE: 'error.txt'
SIM_FILE: 'image_sim.txt'
AUDIO:
RESULT_FILE: 'audio.txt'
OPENSMILE_DIR: '/home/jwq/Downloads/opensmile-2.3.0'
DATA_NAME: 'val.npy'
CLASS:
RESULT_FILE: 'class.txt'
DATA_NAME: 'val _reannotation.npy'
POSE:
TEST_SEGMENT: 8
TEST_CROP: 1
BATCH_SIZE: 1
INPUT_SIZE: 224
MODALITY: 'RGB'
ARCH: 'resnet50'
RESULT_FILE: 'pose_filter.txt'
BG:
RESULT_FILE: 'bg_filter.txt'
DATA_NAME: 'bg_val_feature.npy'
PERSON:
RESULT_FILE: 'person_filter.txt'
DATA_NAME: 'person_val_feature.npy'