Spaces:
Runtime error
Runtime error
File size: 1,570 Bytes
cd3346a | 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 | import json
import os
from glob import glob
from pathlib import Path
def get_original_video_paths(root_dir, basename=False):
originals = set()
originals_v = set()
for json_path in glob(os.path.join(root_dir, "*/metadata.json")):
dir = Path(json_path).parent
with open(json_path, "r") as f:
metadata = json.load(f)
for k, v in metadata.items():
original = v.get("original", None)
if v["label"] == "REAL":
original = k
originals_v.add(original)
originals.add(os.path.join(dir, original))
originals = list(originals)
originals_v = list(originals_v)
print(len(originals))
return originals_v if basename else originals
def get_original_with_fakes(root_dir):
pairs = []
for json_path in glob(os.path.join(root_dir, "*/metadata.json")):
with open(json_path, "r") as f:
metadata = json.load(f)
for k, v in metadata.items():
original = v.get("original", None)
if v["label"] == "FAKE":
pairs.append((original[:-4], k[:-4] ))
return pairs
def get_originals_and_fakes(root_dir):
originals = []
fakes = []
for json_path in glob(os.path.join(root_dir, "*/metadata.json")):
with open(json_path, "r") as f:
metadata = json.load(f)
for k, v in metadata.items():
if v["label"] == "FAKE":
fakes.append(k[:-4])
else:
originals.append(k[:-4])
return originals, fakes
|