From 155a1130b89365b7b604baf8910a126ebc9db5b2 Mon Sep 17 00:00:00 2001 From: Sebastian Fellner Date: Mon, 14 Oct 2019 15:44:19 +0200 Subject: [PATCH] update DATASET_PATHS to reflect changes in download script change 'c0' compression to 'rawy' --- dataset/extract_compressed_videos.py | 11 +++++++---- 1 file changed, 7 insertions(+), 4 deletions(-) diff --git a/dataset/extract_compressed_videos.py b/dataset/extract_compressed_videos.py index c644a40..3b42e91 100644 --- a/dataset/extract_compressed_videos.py +++ b/dataset/extract_compressed_videos.py @@ -15,12 +15,15 @@ DATASET_PATHS = { - 'original': 'original_sequences', + 'original': 'original_sequences/youtube', + 'DeepFakeDetection_original': 'original_sequences/actors', 'Deepfakes': 'manipulated_sequences/Deepfakes', + 'DeepFakeDetection': 'manipulated_sequences/DeepFakeDetection', 'Face2Face': 'manipulated_sequences/Face2Face', - 'FaceSwap': 'manipulated_sequences/FaceSwap' + 'FaceSwap': 'manipulated_sequences/FaceSwap', + 'NeuralTextures': 'manipulated_sequences/NeuralTextures', } -COMPRESSION = ['c0', 'c23', 'c40'] +COMPRESSION = ['raw', 'c23', 'c40'] def extract_frames(data_path, output_path, method='cv2'): @@ -68,7 +71,7 @@ def extract_method_videos(data_path, dataset, compression): choices=list(DATASET_PATHS.keys()) + ['all'], default='all') p.add_argument('--compression', '-c', type=str, choices=COMPRESSION, - default='c0') + default='raw') args = p.parse_args() if args.dataset == 'all':