-
Notifications
You must be signed in to change notification settings - Fork 98
/
make_dataset.sh
52 lines (47 loc) · 1.87 KB
/
make_dataset.sh
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
#!/usr/bin/env bash
echo ""
echo "-------------------------------------------------"
echo "| Index DFDC dataset |"
echo "-------------------------------------------------"
# put your dfdc source directory path and uncomment the following line
# DFDC_SRC=/your/dfdc/train/split/source/directory
python index_dfdc.py --source $DFDC_SRC
echo ""
echo "-------------------------------------------------"
echo "| Index FF dataset |"
echo "-------------------------------------------------"
# put your ffpp source directory path and uncomment the following line
# FFPP_SRC=/your/ffpp/source/directory
python index_ffpp.py --source $FFPP_SRC
echo ""
echo "-------------------------------------------------"
echo "| Extract faces from DFDC |"
echo "-------------------------------------------------"
# put your source and destination directories and uncomment the following lines
# DFDC_SRC=/your/dfdc/source/folder
# VIDEODF_SRC=/previously/computed/index/path
# FACES_DST=/faces/output/directory
# FACESDF_DST=/faces/df/output/directory
# CHECKPOINT_DST=/tmp/per/video/outputs
python extract_faces.py \
--source $DFDC_SRC \
--videodf $VIDEODF_SRC \
--facesfolder $FACES_DST \
--facesdf $FACESDF_DST \
--checkpoint $CHECKPOINT_DST
echo ""
echo "-------------------------------------------------"
echo "| Extract faces from FF |"
echo "-------------------------------------------------"
# put your source and destination directories and uncomment the following lines
# FFPP_SRC=/your/dfdc/source/folder
# VIDEODF_SRC=/previously/computed/index/path
# FACES_DST=/faces/output/directory
# FACESDF_DST=/faces/df/output/directory
# CHECKPOINT_DST=/tmp/per/video/outputs
python extract_faces.py \
--source $FFPP_SRC \
--videodf $VIDEODF_SRC \
--facesfolder $FACES_DST \
--facesdf $FACESDF_DST \
--checkpoint $CHECKPOINT_DST