|
a |
|
b/tools/data/kinetics/extract_frames.sh |
|
|
1 |
#!/usr/bin/env bash |
|
|
2 |
|
|
|
3 |
DATASET=$1 |
|
|
4 |
if [ "$DATASET" == "kinetics400" ] || [ "$1" == "kinetics600" ] || [ "$1" == "kinetics700" ]; then |
|
|
5 |
echo "We are processing $DATASET" |
|
|
6 |
else |
|
|
7 |
echo "Bad Argument, we only support kinetics400, kinetics600 or kinetics700" |
|
|
8 |
exit 0 |
|
|
9 |
fi |
|
|
10 |
|
|
|
11 |
cd ../ |
|
|
12 |
python build_rawframes.py ../../data/${DATASET}/videos_train/ ../../data/${DATASET}/rawframes_train/ --level 2 --flow-type tvl1 --ext mp4 --task both --new-short 256 |
|
|
13 |
echo "Raw frames (RGB and tv-l1) Generated for train set" |
|
|
14 |
|
|
|
15 |
python build_rawframes.py ../../data/${DATASET}/videos_val/ ../../data/${DATASET}/rawframes_val/ --level 2 --flow-type tvl1 --ext mp4 --task both --new-short 256 |
|
|
16 |
echo "Raw frames (RGB and tv-l1) Generated for val set" |
|
|
17 |
|
|
|
18 |
cd ${DATASET}/ |