-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathconfig.py
101 lines (89 loc) · 6.12 KB
/
config.py
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
# *_*coding:utf-8 *_*
import os
import sys
import socket
############ For LINUX ##############
DATA_DIR = {
# 'Track1_English_Annotation': r'/home/zongtianyu/data/wangyuanxiang/xj/track1/English/Annotation',
# 'Track1_English_NoAnnotation': r'/home/zongtianyu/data/wangyuanxiang/xj/track1/English/NoAnnotation',
# 'Track1_Mandarin_Annotation': r'/home/zongtianyu/data/wangyuanxiang/xj/track1/Mandarin/Annotation',
# 'Track1_Mandarin_NoAnnotation': r'/home/zongtianyu/data/wangyuanxiang/xj/track1/Mandarin/NoAnnotation',
'Track2_English': r'/home/zongtianyu/data/wangyuanxiang/xyy/ChallengeData/Track2/English',
'Track2_Mandarin': r'/home/zongtianyu/data/wangyuanxiang/xyy/ChallengeData/Track2/Mandarin',
}
PATH_TO_RAW_AUDIO = {
# 'Track1_English_Annotation': os.path.join(DATA_DIR['Track1_English_Annotation'], 'Audios'),
# 'Track1_English_NoAnnotation': os.path.join(DATA_DIR['Track1_English_NoAnnotation'], 'Audios'),
# 'Track1_Mandarin_Annotation': os.path.join(DATA_DIR['Track1_Mandarin_Annotation'], 'Audios'),
# 'Track1_Mandarin_NoAnnotation': os.path.join(DATA_DIR['Track1_Mandarin_NoAnnotation'], 'Audios'),
'Track2_English': os.path.join('/home/zongtianyu/data/wangyuanxiang/xyy/ChallengeData/Track2/English/', 'Testing_Audios_resampled_16000'),
'Track2_Mandarin': os.path.join(DATA_DIR['Track2_Mandarin'], 'Audios'),
}
PATH_TO_RAW_FACE = {
# 'MER2023': r'G:\数据集\MELD.Raw\train_openface' #os.path.join(DATA_DIR['MER2023'], 'openface_face'),
# 'Track1_English_Annotation': os.path.join(DATA_DIR['Track1_English_Annotation'], 'openface_face'),
# 'Track1_English_NoAnnotation': os.path.join(DATA_DIR['Track1_English_NoAnnotation'], 'openface_face'),
# 'Track1_Mandarin_Annotation': os.path.join(DATA_DIR['Track1_Mandarin_Annotation'], 'openface_face'),
# 'Track1_Mandarin_NoAnnotation': os.path.join(DATA_DIR['Track1_Mandarin_NoAnnotation'], 'openface_face'),
'Track2_English': os.path.join(DATA_DIR['Track2_English'], 'Testing_video_face_frames'),
'Track2_Mandarin': os.path.join(DATA_DIR['Track2_Mandarin'], 'openface_face'),
}
PATH_TO_TRANSCRIPTIONS = {
# 'MER2023': os.path.join(DATA_DIR['MER2023'], 'transcription.csv'),
# 'Track1_English_Annotation': os.path.join(DATA_DIR['Track1_English_Annotation'], 'transcription.csv'),
# 'Track1_English_NoAnnotation': os.path.join(DATA_DIR['Track1_English_NoAnnotation'], 'transcription.csv'),
# 'Track1_Mandarin_Annotation': os.path.join(DATA_DIR['Track1_Mandarin_Annotation'], 'transcription.csv'),
# 'Track1_Mandarin_NoAnnotation': os.path.join(DATA_DIR['Track1_Mandarin_NoAnnotation'], 'transcription.csv'),
'Track2_English': os.path.join(DATA_DIR['Track2_English'], 'Training_transcription.csv'),
'Track2_Mandarin': os.path.join(DATA_DIR['Track2_Mandarin'], 'transcription.csv'),
}
PATH_TO_FEATURES = {
# 'MER2023': r'G:\数据集\IEMOCAP\RoBERTa' #os.path.join(DATA_DIR['MER2023'], 'features'),
# 'Track1_English_Annotation': os.path.join(DATA_DIR['Track1_English_Annotation'], 'features'),
# 'Track1_English_NoAnnotation': os.path.join(DATA_DIR['Track1_English_NoAnnotation'], 'features'),
# 'Track1_Mandarin_Annotation': os.path.join(DATA_DIR['Track1_Mandarin_Annotation'], 'features'),
# 'Track1_Mandarin_NoAnnotation': os.path.join(DATA_DIR['Track1_Mandarin_NoAnnotation'], 'features'),
'Track2_English': os.path.join(DATA_DIR['Track2_English'], 'features'),
'Track2_Mandarin': os.path.join(DATA_DIR['Track2_Mandarin'], 'features'),
}
# PATH_TO_LABEL = {
# 'MER2023': os.path.join(DATA_DIR['MER2023'], 'label-6way.npz'),
# }
PATH_TO_PRETRAINED_MODELS = r'/home/zongtianyu/data/wangyuanxiang/xyy/models/'
PATH_TO_OPENSMILE = os.path.join(PATH_TO_PRETRAINED_MODELS, r'opensmile-2.3.0')
PATH_TO_FFMPEG = os.path.join(PATH_TO_PRETRAINED_MODELS, r'ffmpeg-4.4.1-i686-static', r'ffmpeg')
PATH_TO_NOISE = os.path.join(PATH_TO_PRETRAINED_MODELS, r'musan', r'audio-select')
SAVED_ROOT = os.path.join('./saved')
DATA_DIR = os.path.join(SAVED_ROOT, 'data')
MODEL_DIR = os.path.join(SAVED_ROOT, 'model')
LOG_DIR = os.path.join(SAVED_ROOT, 'log')
PREDICTION_DIR = os.path.join(SAVED_ROOT, 'prediction')
FUSION_DIR = os.path.join(SAVED_ROOT, 'fusion')
SUBMISSION_DIR = os.path.join(SAVED_ROOT, 'submission')
PATH_TO_OPENFACE = r"/home/zongtianyu/data/wangyuanxiang/xyy/tools/OpenFace_2.2.0/exe/"
############ For Windows (openface-win) ##############
# DATA_DIR_Win = {
# 'Track1_English_Annotation': r'G:\数据集\ChallengeData\Track1\English\Annotation',
# 'Track1_English_NoAnnotation': r'G:\数据集\ChallengeData\Track1\English\NoAnnotation',
# 'Track1_Mandarin_Annotation': r'G:\数据集\ChallengeData\Track1\Mandarin\Annotation',
# 'Track1_Mandarin_NoAnnotation': r'G:\数据集\ChallengeData\Track1\Mandarin\NoAnnotation',
# 'Track2_English': r'G:\数据集\ChallengeData\Track2\English',
# 'Track2_Mandarin': r'G:\数据集\ChallengeData\Track2\Mandarin',
# }
# PATH_TO_RAW_FACE_Win = {
# 'Track1_English_Annotation': os.path.join(DATA_DIR_Win['Track1_English_Annotation'], 'openface_face'),
# 'Track1_English_NoAnnotation': os.path.join(DATA_DIR_Win['Track1_English_NoAnnotation'], 'openface_face'),
# 'Track1_Mandarin_Annotation': os.path.join(DATA_DIR_Win['Track1_Mandarin_Annotation'], 'openface_face'),
# 'Track1_Mandarin_NoAnnotation': os.path.join(DATA_DIR_Win['Track1_Mandarin_NoAnnotation'], 'openface_face'),
# 'Track2_English': os.path.join(DATA_DIR_Win['Track2_English'], 'openface_face'),
# 'Track2_Mandarin': os.path.join(DATA_DIR_Win['Track2_Mandarin'], 'openface_face'),
# }
# PATH_TO_FEATURES_Win = {
# 'Track1_English_Annotation': os.path.join(DATA_DIR_Win['Track1_English_Annotation'], 'features'),
# 'Track1_English_NoAnnotation': os.path.join(DATA_DIR_Win['Track1_English_NoAnnotation'], 'features'),
# 'Track1_Mandarin_Annotation': os.path.join(DATA_DIR_Win['Track1_Mandarin_Annotation'], 'features'),
# 'Track1_Mandarin_NoAnnotation': os.path.join(DATA_DIR_Win['Track1_Mandarin_NoAnnotation'], 'features'),
# 'Track2_English': os.path.join(DATA_DIR_Win['Track2_English'], 'features'),
# 'Track2_Mandarin': os.path.join(DATA_DIR_Win['Track2_Mandarin'], 'features'),
# }
# PATH_TO_OPENFACE_Win = r"E:\Project\Zuohaolin\MER2023-Baseline-master\tools\OpenFace_2.2.0_win_x64"