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':