-
Notifications
You must be signed in to change notification settings - Fork 2
Expand file tree
/
Copy pathfilePath.py
More file actions
59 lines (47 loc) · 2.03 KB
/
filePath.py
File metadata and controls
59 lines (47 loc) · 2.03 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
# -*- coding: UTF-8 -*-
import os
# The path to the downloaded DeepFake video datasets
original_videos_path = {
'UADFV': './datasets/Videos/UADFV/',
'DeepfakeTIMIT_HQ': './datasets/Videos/DeepfakeTIMIT_frames/',
'DeepfakeTIMIT_LQ': './datasets/Videos/DeepfakeTIMIT_frames/', # provided frames
'FF++_Deepfakes_c40': './datasets/Videos/FF++/Deepfakes/c40/',
'FF++_Deepfakes_raw': './datasets/Videos/FF++/Youtube/raw/',
'DFD': './datasets/Videos/DFD/',
'DFDC': './datasets/Videos/DFDC/',
'CelebDF_V2': './datasets/Videos/CelebDF_V2/',
}
# The path of the videos divided into the training set and the test set.
split_path = {
'UADFV': './datasets/video_datasets/UADFV/',
'DeepfakeTIMIT': './datasets/video_datasets/DeepfakeTIMIT_frames/',
'DeepfakeTIMIT_LQ': './datasets/video_datasets/DeepfakeTIMIT_LQ_frames/',
'FF++_Deepfakes': './datasets/video_datasets/FF++_Deepfakes/',
'FF++_Deepfakes_raw': './datasets/video_datasets/FF++_Deepfakes_raw/',
'FF++_Face2Face': './datasets/video_datasets/FF++_Face2Face/',
'FF++_FaceSwap': './datasets/video_datasets/FF++_FaceSwap/',
'FF++_NeuralTextures': './datasets/video_datasets/FF++_NeuralTextures/',
'DFD': './datasets/video_datasets/DFD/raw/',
'DFDC': './datasets/video_datasets/DFDC/',
'CelebDF_V2': './datasets/video_datasets/CelebDF_V2/',
}
train_dir = {}
test_dir = {}
for f in split_path.keys():
train_dir[f] = os.path.join(split_path[f], 'train')
test_dir[f] = os.path.join(split_path[f], 'test')
train_real_dir = {}
train_fake_dir = {}
validation_real_dir = {}
validation_fake_dir = {}
test_real_dir = {}
test_fake_dir = {}
for f in split_path.keys():
train_real_dir[f] = os.path.join(train_dir[f], 'real')
train_fake_dir[f] = os.path.join(train_dir[f], 'fake')
test_real_dir[f] = os.path.join(test_dir[f], 'real')
test_fake_dir[f] = os.path.join(test_dir[f], 'fake')
# The path to the extracted frame
frames_path = './datasets/frames_datasets/'
# The path to the extracted faces
faces_path = './datasets/faces/'